From 0bd99bbe9a19410992c4d9c5ce76c0890d41850a Mon Sep 17 00:00:00 2001 From: spencerfolk Date: Tue, 2 Jan 2024 15:21:28 -0500 Subject: [PATCH] Fixed model save/log names --- examples/ppo_hover_train.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/examples/ppo_hover_train.py b/examples/ppo_hover_train.py index 14359e9..30d7368 100644 --- a/examples/ppo_hover_train.py +++ b/examples/ppo_hover_train.py @@ -74,9 +74,9 @@ epoch_count = 0 while True: # Run indefinitely.. # This line will run num_timesteps for training and log the results every so often. - model.learn(total_timesteps=num_timesteps, reset_num_timesteps=False, tb_log_name="PPO-Quad"+start_time.strftime('%H-%M-%S')) + model.learn(total_timesteps=num_timesteps, reset_num_timesteps=False, tb_log_name="PPO-Quad_"+start_time.strftime('%H-%M-%S')) # Save the model - model.save(f"{models_dir}/PPO_hover_{num_timesteps*(epoch_count+1)}") + model.save(f"{models_dir}/PPO_hover_policy_{num_timesteps*(epoch_count+1)}") epoch_count += 1 \ No newline at end of file