From 7415b329d114e1043b77eedb59b1dae35ae9cb53 Mon Sep 17 00:00:00 2001 From: Haichao Zhang Date: Fri, 13 Sep 2024 14:03:59 -0700 Subject: [PATCH] Fix the issue that supervised algorithm does not produce tb summary curves --- alf/trainers/policy_trainer.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/alf/trainers/policy_trainer.py b/alf/trainers/policy_trainer.py index 72fd6d1df..d4547a205 100644 --- a/alf/trainers/policy_trainer.py +++ b/alf/trainers/policy_trainer.py @@ -828,7 +828,8 @@ def __init__(self, config: TrainerConfig): self._num_epochs = config.num_iterations self._trainer_progress.set_termination_criterion(self._num_epochs) - self._algorithm = config.algorithm_ctor(config=config) + self._algorithm = config.algorithm_ctor( + config=config, debug_summaries=self._debug_summaries) self._algorithm.set_path('') def _train(self):