def from_flags(restore_dir, log_dir, restore_continue, restore_itr,
restart_at_zero=False, strict='y'):
if restore_dir is None:
return None
strict = {'y': True, 'n': False}[strict]
return TrainRestorer(get_ckpts_dir(get_experiment_dir(log_dir, restore_dir)),
restore_continue, restore_itr, restart_at_zero, strict)
def get_configs_experiment_dir(prefix, log_dir, log_date):
experiment_dir = paths.get_experiment_dir(log_dir, log_date)
log_dir_comps = logdir_helpers.parse_log_dir(experiment_dir,
DEFAULT_CONFIG_DIR,
[prefix, 'dl'],
append_ext='.cf')
config_ps = log_dir_comps.config_paths
global_config.add_from_flag(log_dir_comps.postfix)
return config_ps, experiment_dir
def from_flags(restore_dir,
log_dir,
restore_continue,
restore_itr,
restart_at_zero=False,
strict='y'):
if restore_dir is None:
return None
strict = {'y': True, 'n': False}[strict]
# TODO
for postfix in ('', '/AWS'):
try:
experiment_dir = get_experiment_dir(log_dir + postfix,
restore_dir)
break
except ValueError as e:
print(e)
else: # no-break
raise ValueError('Not found!', restore_dir, log_dir)
return TrainRestorer(get_ckpts_dir(experiment_dir), restore_continue,
restore_itr, restart_at_zero, strict)