Changed naming scheme

2024-01-03 23:41:24 -05:00
parent 1769323890
commit 1b634ee1af
1 changed files with 1 additions and 1 deletions
--- a/examples/ppo_hover_train.py
+++ b/examples/ppo_hover_train.py
@@ -77,6 +77,6 @@ while True:  # Run indefinitely..
    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_policy_{num_timesteps*(epoch_count+1)}")
+    model.save(f"{models_dir}/PPO/{start_time.strftime('%H-%M-%S')}/hover_{num_timesteps*(epoch_count+1)}")

    epoch_count += 1