gaspersic/baselines
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

ncpu needs to be an integer. (#558)

Browse Source
This commit is contained in:
uronce-cc
2018-09-01 00:02:18 +08:00
committed by pzhokhov
parent 1937826784
commit 0a40206c6c
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
baselines/run.py
View File

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