swangnice/rotor_py_control
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Fixed model save/log names

Browse Source
This commit is contained in:
spencerfolk
2024-01-02 15:21:28 -05:00
parent 4e4c361289
commit 0bd99bbe9a
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
examples/ppo_hover_train.py
View File

@@ -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