def _get_restore_ckpt_dir(restore_flag):
if Saver.is_ckpt_dir(restore_flag):
return restore_flag
if Saver.is_ckpt_dir(Saver.ckpt_dir_for_log_dir(restore_flag)):
return Saver.ckpt_dir_for_log_dir(restore_flag)
raise ValueError('Invalid ckpt dir: {}'.format(restore_flag))
