fix bug

alibaba · Jul 19, 2023 · a380eee · a380eee
1 parent eb535f3
commit a380eee
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/easy_rec/python/main.py b/easy_rec/python/main.py
@@ -366,7 +366,8 @@ def _train_and_evaluate_impl(pipeline_config,
         input_fn=eval_input_fn,
         max_steps=fit_on_eval_steps,
         hooks=list(train_spec.hooks),
-        saving_listeners=train_spec.saving_listeners)
+        saving_listeners=train_spec.saving_listeners if hasattr(
+            train_spec, 'saving_listeners') else None)
     logging.info('Finished training on eval data')
   # return estimator for custom training using estimator.train
   return estimator