Skip to content

S
SCL-my
Commit 40d3539c by yuxguo
Options

fix

parent 67498b6b
Showing with 1 additions and 1 deletions
main.py
...@@ -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