Update ray gpu

hannanabdul55 · Mar 9, 2021 · d388a64 · d388a64
1 parent 00b059a
commit d388a64
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/experiment/experiment_s_p.py b/experiment/experiment_s_p.py
@@ -56,6 +56,7 @@
     kwargs['num_gpus'] = n_gpus
     exp_kwargs['num_gpus'] = 1
 else:
+    exp_kwargs['num_gpus'] = 0
     n_gpus = 0
 
 if args.workers: