From ea9d8dc00950da8f6c22b5d66e987478ef55108f Mon Sep 17 00:00:00 2001 From: spencerfolk Date: Thu, 4 Jan 2024 18:17:43 -0500 Subject: [PATCH] Bug fix in example --- examples/ppo_hover_eval.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/examples/ppo_hover_eval.py b/examples/ppo_hover_eval.py index 19a70dc..5bf5ea1 100644 --- a/examples/ppo_hover_eval.py +++ b/examples/ppo_hover_eval.py @@ -109,7 +109,7 @@ num_timesteps_idxs = [int(input("Enter the epoch index: "))] # Evaluation... for (k, num_timesteps_idx) in enumerate(num_timesteps_idxs): # For each num_timesteps index... - print(f"[ppo_hover_eval.py]: Starting epoch {k+1} out of {len(epoch_idxs)}.") + print(f"[ppo_hover_eval.py]: Starting epoch {k+1} out of {len(num_timesteps_idxs)}.") # Load the model for the appropriate epoch. model_path = os.path.join(num_timesteps_dir, num_timesteps_list_sorted[num_timesteps_idx]) @@ -117,7 +117,7 @@ for (k, num_timesteps_idx) in enumerate(num_timesteps_idxs): # For each num_tim model = PPO.load(model_path, env=envs[0], tensorboard_log=log_dir) # Set figure title for 3D plot. - fig.suptitle(f"Model: PPO/{models_available[model_idx]}, Epoch: {extract_number(num_timesteps_list_sorted[num_timesteps_idx]):,}") + fig.suptitle(f"Model: PPO/{models_available[model_idx]}, Num Timesteps: {extract_number(num_timesteps_list_sorted[num_timesteps_idx]):,}") # Visualization is slow, so we'll also save frames to make a GIF later. # Set the path for these frames here.