From 36aca1ce70478d9f769f805c315b372fbea3bb11 Mon Sep 17 00:00:00 2001 From: NickDixon37 <82039606+NickDixon37@users.noreply.github.com> Date: Thu, 7 Nov 2024 15:09:32 -0600 Subject: [PATCH] Update run_baseline_parallel_fast.py added argument for line 51 --- baselines/run_baseline_parallel_fast.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/baselines/run_baseline_parallel_fast.py b/baselines/run_baseline_parallel_fast.py index 076e51c0..8a3177cb 100644 --- a/baselines/run_baseline_parallel_fast.py +++ b/baselines/run_baseline_parallel_fast.py @@ -48,7 +48,7 @@ def _init(): checkpoint_callback = CheckpointCallback(save_freq=ep_length, save_path=sess_path, name_prefix='poke') - callbacks = [checkpoint_callback, TensorboardCallback()] + callbacks = [checkpoint_callback, TensorboardCallback(log_dir=sess_path)] if use_wandb_logging: import wandb