def train(load_ckpt=None):
if load_ckpt is None:
args = [
'--n_workers=8', '--steps_per_update=30', 'AI4U-v0',
'--n_steps=10e8'
]
else:
args = [
'--n_workers=8', '--steps_per_update=30', 'AI4U-v0',
'--load_ckpt=%s' % (load_ckpt)
]
make_env_def()
run_train(environment_definitions, args)
def train():
args = ['--n_workers=8', '--steps_per_update=30', 'AI4U-v0']
make_env_def()
run_train(environment_definitions, args)
def train():
args = ['--n_workers=4', 'AI4U-v0']
make_env_def()
run_train(environment_definitions, args)