Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Adds resuming and saving model to sb3 example #135

Merged
merged 10 commits into from
Jul 23, 2023
Prev Previous commit
Add default=False to inference
Co-authored-by: Florentin Luca Rieger <[email protected]>
Ivan-267 and visuallization authored Jul 23, 2023

Verified

This commit was created on GitHub.com and signed with GitHub’s verified signature. The key has expired.
commit 82bd742c1b30a184329548a17673b108098fe271
1 change: 1 addition & 0 deletions examples/stable_baselines3_example.py
Original file line number Diff line number Diff line change
@@ -72,6 +72,7 @@
)
parser.add_argument(
"--inference",
Ivan-267 marked this conversation as resolved.
Show resolved Hide resolved
default=False,
action="store_true",
help="Instead of training, it will run inference on a loaded model for --timesteps steps. "
"Requires --resume_model_path to be set."