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