Skip to content

Commit b0269f8

Browse files
committed
fix TFModel.evaluate() not to create saver when it exists
1 parent faaef36 commit b0269f8

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

tensorflow_model.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -123,7 +123,8 @@ def evaluate(self) -> Optional[ModelEvaluationResults]:
123123

124124
self.eval_top_words_op, self.eval_top_values_op, self.eval_original_names_op, _, _, _, _, \
125125
self.eval_code_vectors = self._build_tf_test_graph(input_tensors)
126-
self.saver = tf.compat.v1.train.Saver()
126+
if self.saver is None:
127+
self.saver = tf.compat.v1.train.Saver()
127128

128129
if self.config.MODEL_LOAD_PATH and not self.config.TRAIN_DATA_PATH_PREFIX:
129130
self._initialize_session_variables()

0 commit comments

Comments
 (0)