Commit bff69e84 authored by Yuxin Wu's avatar Yuxin Wu

update varnames

parent 58a41fca
......@@ -62,7 +62,7 @@ def run_test(path, input):
input_var_names=['input'],
session_init=ParamRestore(param_dict),
session_config=get_default_sess_config(0.9),
output_var_names=['output:0'] # output:0 is the probability distribution
output_var_names=['output'] # output:0 is the probability distribution
)
predict_func = get_predict_func(pred_config)
......
......@@ -79,7 +79,7 @@ def run_test(path, input):
input_var_names=['input'],
session_init=ParamRestore(param_dict),
session_config=get_default_sess_config(0.9),
output_var_names=['output:0'] # output:0 is the probability distribution
output_var_names=['output'] # output:0 is the probability distribution
)
predict_func = get_predict_func(pred_config)
......
......@@ -67,7 +67,8 @@ def set_logger_dir(dirname, action=None):
if os.path.isdir(dirname):
if not action:
_logger.warn("""\
Directory {} exists! Please either backup/delete it, or use a new directory. \
Directory {} exists! Please either backup/delete it, or use a new directory.""")
_logger.warn("""\
If you're resuming from a previous run you can choose to keep it.""".format(dirname))
_logger.info("Select Action: k (keep) / b (backup) / d (delete) / n (new):")
while not action:
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment