diff --git a/examples/sb3_imitation.py b/examples/sb3_imitation.py index 75b1b945..16925afa 100644 --- a/examples/sb3_imitation.py +++ b/examples/sb3_imitation.py @@ -237,10 +237,8 @@ def close_env(): print(f"Mean reward after evaluation: {mean_reward}") except (KeyboardInterrupt, ConnectionError, ConnectionResetError): - print( - """Training interrupted by user or a ConnectionError. Will save if --save_model_path was - used and/or export if --onnx_export_path was used.""" - ) + print("""Training interrupted by user or a ConnectionError. Will save if --save_model_path was + used and/or export if --onnx_export_path was used.""") finally: handle_onnx_export() diff --git a/examples/stable_baselines3_example.py b/examples/stable_baselines3_example.py index fc4ef98c..b2c3e4a4 100644 --- a/examples/stable_baselines3_example.py +++ b/examples/stable_baselines3_example.py @@ -100,6 +100,12 @@ default=False, ) parser.add_argument("--speedup", default=1, type=int, help="Whether to speed up the physics in the env") +parser.add_argument( + "--action_repeat", + default=None, + type=int, + help="Similar concept to frame skip, sends action/gets obs every n frames only. Uses sync node setting if not set here.", +) parser.add_argument( "--n_parallel", default=1, @@ -158,7 +164,12 @@ def cleanup(): print("Info: Using --viz without --env_path set has no effect, in-editor training will always render.") env = StableBaselinesGodotEnv( - env_path=args.env_path, show_window=args.viz, seed=args.seed, n_parallel=args.n_parallel, speedup=args.speedup + env_path=args.env_path, + show_window=args.viz, + seed=args.seed, + n_parallel=args.n_parallel, + speedup=args.speedup, + action_repeat=args.action_repeat, ) env = VecMonitor(env) @@ -220,9 +231,7 @@ def func(progress_remaining: float) -> float: try: model.learn(**learn_arguments) except (KeyboardInterrupt, ConnectionError, ConnectionResetError): - print( - """Training interrupted by user or a ConnectionError. Will save if --save_model_path was - used and/or export if --onnx_export_path was used.""" - ) + print("""Training interrupted by user or a ConnectionError. Will save if --save_model_path was + used and/or export if --onnx_export_path was used.""") finally: cleanup()