amend

pytorch · Oct 3, 2023 · d7bb731 · d7bb731
1 parent 096d622
commit d7bb731
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/examples/ddpg/ddpg.py b/examples/ddpg/ddpg.py
@@ -183,8 +183,8 @@ def main(cfg: "DictConfig"):  # noqa: F821
                 eval_reward = eval_rollout["next", "reward"].sum(-2).mean().item()
                 metrics_to_log["eval/reward"] = eval_reward
                 metrics_to_log["eval/time"] = eval_time
-
-        log_metrics(logger, metrics_to_log, collected_frames)
+        if logger is not None:
+            log_metrics(logger, metrics_to_log, collected_frames)
         sampling_start = time.time()
 
     collector.shutdown()