Commit 762521e1 authored by Yuxin Wu's avatar Yuxin Wu

fix #gpu in cifar example

parent 23c36186
......@@ -150,7 +150,7 @@ if __name__ == '__main__':
if args.load:
config.session_init = SaverRestore(args.load)
config.nr_tower = get_nr_gpu()
config.nr_tower = max(get_nr_gpu(), 1)
if config.nr_tower <= 1:
QueueInputTrainer(config).train()
else:
......
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