Commit 40d3539c by yuxguo

fix

parent 67498b6b
...@@ -454,7 +454,7 @@ def main(): ...@@ -454,7 +454,7 @@ def main():
model = get_model(args) model = get_model(args)
if args.use_gpu: if args.use_gpu:
model = DataParallel(model).cuda() model = DataParallel(model, device_ids=[0, 1]).cuda()
# model = model.cuda() # model = model.cuda()
if args.weight_decay == 0: if args.weight_decay == 0:
......
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