丁云鹏 5 месяцев назад
Родитель
Сommit
f7ae66cdfa

+ 0 - 8
recommend-model-produce/src/main/python/tools/static_ps_trainer_v2.py

@@ -266,14 +266,6 @@ class Main(object):
                                program.global_block().vars.keys()))
                 else:
                     feedvars.append(program.global_block().vars[var_name])
-            for var_name in fetch_targets:
-                if var_name not in program.global_block().vars:
-                    raise ValueError(
-                        "Fetch variable: {} not in default_main_program, global block has follow vars: {}".
-                        format(var_name,
-                               program.global_block().vars.keys()))
-                else:
-                    fetchvars.append(program.global_block().vars[var_name])
 
             paddle.static.save_inference_model(
                 os.path.join(model_dir, "dnn_plugin_new"),