Ejemplo n.º 1
0
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)
Ejemplo n.º 3
0
def train():
    args = ['--n_workers=4', 'AI4U-v0']
    make_env_def()
    run_train(environment_definitions, args)