ncpu needs to be an integer. (#558)

2018-09-01 00:02:18 +08:00
parent 1937826784
commit 0a40206c6c
1 changed files with 1 additions and 1 deletions
--- a/baselines/run.py
+++ b/baselines/run.py
@@ -74,7 +74,7 @@ def train(args, extra_args):

 def build_env(args):
    ncpu = multiprocessing.cpu_count()
-    if sys.platform == 'darwin': ncpu /= 2
+    if sys.platform == 'darwin': ncpu //= 2
    nenv = args.num_env or ncpu
    alg = args.alg
    rank = MPI.COMM_WORLD.Get_rank() if MPI else 0