File tree 1 file changed +3
-1
lines changed
1 file changed +3
-1
lines changed Original file line number Diff line number Diff line change @@ -895,7 +895,7 @@ def save_inference_model(dirname,
895
895
True is supported.
896
896
897
897
Returns:
898
- None
898
+ target_var_name_list(list): The fetch variables' name list
899
899
900
900
Raises:
901
901
ValueError: If `feed_var_names` is not a list of basestring.
@@ -954,6 +954,7 @@ def save_inference_model(dirname,
954
954
var , 1. , name = "save_infer_model/scale_{}" .format (i ))
955
955
uniq_target_vars .append (var )
956
956
target_vars = uniq_target_vars
957
+ target_var_name_list = [var .name for var in target_vars ]
957
958
958
959
# when a pserver and a trainer running on the same machine, mkdir may conflict
959
960
try :
@@ -1010,6 +1011,7 @@ def save_inference_model(dirname,
1010
1011
params_filename = os .path .basename (params_filename )
1011
1012
1012
1013
save_persistables (executor , dirname , main_program , params_filename )
1014
+ return target_var_name_list
1013
1015
1014
1016
1015
1017
def load_inference_model (dirname ,
You can’t perform that action at this time.
0 commit comments