def run(blueoil_config_file, experiment_id):
"""Train from blueoil config.
Args:
blueoil_config_file:
experiment_id:
"""
if horovod_util.is_enabled():
horovod_util.setup()
if horovod_util.is_rank0():
# Copy bueoil config yaml.
output_dir = os.environ.get('OUTPUT_DIR', 'saved')
experiment_dir = os.path.join(output_dir, experiment_id)
save_config_file(blueoil_config_file, experiment_dir)
# Generete lmnet config from blueoil config.
# this lmnet_config_file cannot be reuse from multiprocesses as the file is a named temporary file.
lmnet_config_file = generate(blueoil_config_file)
# Start training
run_train(network=None,
dataset=None,
config_file=lmnet_config_file,
experiment_id=experiment_id,
recreate=False)
def run(blueoil_config_file, experiment_id):
"""Train from blueoil config."""
# Generete lmnet config from blueoil config.
lmnet_config_file = generate(blueoil_config_file)
# Start training
run_train(network=None,
dataset=None,
config_file=lmnet_config_file,
experiment_id=experiment_id,
recreate=False)
def run(blueoil_config_file, experiment_id):
"""Train from blueoil config."""
# Copy bueoil config yaml.
output_dir = os.environ.get('OUTPUT_DIR', 'saved')
experiment_dir = os.path.join(output_dir, experiment_id)
save_config_file(blueoil_config_file, experiment_dir)
# Generete lmnet config from blueoil config.
lmnet_config_file = generate(blueoil_config_file)
# Start training
run_train(network=None, dataset=None, config_file=lmnet_config_file, experiment_id=experiment_id, recreate=False)