Skip to content

P
prenet
Commit 7a41852c authored by lishen's avatar lishen
Browse files
Options

[fix]

parent 1980d3eb
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
main.py
View file @ 7a41852c
...@@ -213,7 +213,7 @@ def main(): ...@@ -213,7 +213,7 @@ def main():
cudnn.benchmark = True cudnn.benchmark = True
net.cuda() net.cuda()
device_ids = [0, 1] device_ids = [0]
# net = nn.DataParallel(net).to(device_ids) # net = nn.DataParallel(net).to(device_ids)
net = nn.DataParallel(net, device_ids=device_ids) net = nn.DataParallel(net, device_ids=device_ids)
# optimizer = nn.DataParallel(optimizer, device_ids=device_ids) # optimizer = nn.DataParallel(optimizer, device_ids=device_ids)
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment