diff --git a/README.md b/README.md index f85c33d9..dcf55115 100644 --- a/README.md +++ b/README.md @@ -169,6 +169,7 @@ pyright src/agent ## TODO - [x] Address CI errors -- [ ] Add coding guidelines +- [x] Add coding guidelines - [x] Add teleop interface and examples - [x] Add BC algorithms +- [ ] Add more examples diff --git a/examples/run_ppo_gs.py b/examples/run_ppo_gs.py index 07ddb295..28edde89 100644 --- a/examples/run_ppo_gs.py +++ b/examples/run_ppo_gs.py @@ -75,7 +75,7 @@ def main(num_envs: int = 2048, show_viewer: bool = False, train: bool = True) -> print("Training completed successfully!") print(f"Training completed in {train_summary_info['total_time']:.2f} seconds.") - print(f"Total episodes: {train_summary_info['total_episodes']}.") + print(f"Total iterations: {train_summary_info['total_iterations']}.") print(f"Total steps: {train_summary_info['total_steps']}.") print(f"Total reward: {train_summary_info['final_reward']:.2f}.") else: diff --git a/examples/run_ppo_gym.py b/examples/run_ppo_gym.py index c59dd4ab..28b71cea 100644 --- a/examples/run_ppo_gym.py +++ b/examples/run_ppo_gym.py @@ -107,7 +107,7 @@ def main(train: bool = True) -> None: train_summary_info = runner.train(metric_logger=logger) print("Training completed successfully!") print(f"Training completed in {train_summary_info['total_time']:.2f} seconds.") - print(f"Total episodes: {train_summary_info['total_episodes']}.") + print(f"Total iterations: {train_summary_info['total_iterations']}.") print(f"Total steps: {train_summary_info['total_steps']}.") print(f"Total reward: {train_summary_info['final_reward']:.2f}.")