From bce596cd54035389f4053f38e3a507abfc3295ea Mon Sep 17 00:00:00 2001 From: tink2123 Date: Wed, 17 Jun 2020 16:11:29 +0800 Subject: [PATCH] fix can not find tmp_4 --- tools/program.py | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/tools/program.py b/tools/program.py index 30e9d737..3c71065a 100755 --- a/tools/program.py +++ b/tools/program.py @@ -114,7 +114,10 @@ def merge_config(config): global_config[key] = value else: sub_keys = key.split('.') - assert (sub_keys[0] in global_config), "the sub_keys can only be one of global_config: {}, but get: {}, please check your running command".format(global_config.keys(), sub_keys[0]) + assert ( + sub_keys[0] in global_config + ), "the sub_keys can only be one of global_config: {}, but get: {}, please check your running command".format( + global_config.keys(), sub_keys[0]) cur = global_config[sub_keys[0]] for idx, sub_key in enumerate(sub_keys[1:]): assert (sub_key in cur) @@ -177,7 +180,6 @@ def build(config, main_prog, startup_prog, mode): optimizer.minimize(opt_loss) opt_loss_name = opt_loss.name global_lr = optimizer._global_learning_rate() - global_lr.persistable = True fetch_name_list.insert(0, "lr") fetch_varname_list.insert(0, global_lr.name) return (dataloader, fetch_name_list, fetch_varname_list, opt_loss_name)