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

Changed naming scheme

Browse Source
This commit is contained in:
spencerfolk
2024-01-03 23:41:24 -05:00
parent 1769323890
commit 1b634ee1af
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
examples/ppo_hover_train.py
View File

@@ -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')) model.learn(total_timesteps=num_timesteps, reset_num_timesteps=False, tb_log_name="PPO-Quad_"+start_time.strftime('%H-%M-%S'))
# Save the model # 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 epoch_count += 1