No public description

PiperOrigin-RevId: 595902166

Author: tensorflower-gardener

official/projects/mosaic/train.py

@@ -33,6 +33,7 @@
 from official.projects.mosaic import mosaic_tasks
 from official.projects.mosaic import registry_imports as mosaic_registry_imports
 from official.vision import registry_imports
+from official.vision.utils import summary_manager
 # pylint: enable=unused-import
 
 FLAGS = flags.FLAGS
@@ -94,7 +95,11 @@ def main(_):
       mode=FLAGS.mode,
       params=params,
       model_dir=model_dir,
-      trainer=mosaic_trainer)
+      trainer=mosaic_trainer,
+      eval_summary_manager=summary_manager.maybe_build_eval_summary_manager(
+          params=params, model_dir=model_dir
+      ),
+  )
 
   train_utils.save_gin_config(FLAGS.mode, model_dir)