def main(log_dir, config, cpu):
variant = default_test_config
if config:
with open(os.path.join(config)) as f:
exp_params = json.load(f)
overwrite_dict(variant, exp_params)
experiment(log_dir, variant, cpu=cpu)
def main(config, cpu, log_dir, snapshot_gap):
variant = default_train_config
if config:
with open(os.path.join(config)) as f:
exp_params = json.load(f)
overwrite_dict(variant, exp_params)
# Set log directory.
exp_id = get_exp_id(variant)
variant.update(exp_id=exp_id)
log_dir = create_exp_name(os.path.join(log_dir, exp_id))
print('Logging to:', log_dir)
setup_logger(log_dir=log_dir,
variant=variant,
snapshot_mode='gap_and_last',
snapshot_gap=snapshot_gap,
)
# Set GPU.
if not cpu:
ptu.set_gpu_mode(True)
experiment(variant)