-
Notifications
You must be signed in to change notification settings - Fork 5.8k
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
[rllib] stopper support for python based training #29972
Conversation
Signed-off-by: Max Pumperla <[email protected]>
@@ -102,10 +102,12 @@ def run(example_id: str = typer.Argument(..., help="Example ID to run.")): | |||
example_file = get_example_file(example_id) | |||
example_file, temp_file = download_example_file(example_file) | |||
file_type = example.get("file_type", SupportedFileType.yaml) | |||
stop = example.get("stop", "{}") |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Btw, sorry, unrelated to this PR, BUT can we automate the file_type recognition (instead of making it yaml by default, detect type from file extension).
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
done
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Looks great, thanks for the PR.
Please address the questions and nits before merging. Thx @maxpumperla !
Signed-off-by: Max Pumperla <[email protected]>
Signed-off-by: Max Pumperla <[email protected]>
Signed-off-by: Max Pumperla <[email protected]>
Signed-off-by: Max Pumperla <[email protected]>
Signed-off-by: Max Pumperla <[email protected]>
Signed-off-by: Max Pumperla <[email protected]> Signed-off-by: Weichen Xu <[email protected]>
rllib train
python files are now cleaner, split out Tune-specificstop
conditions. So, you can now write:I updated a couple of example scripts / tuned examples to show-case this (and added a test). I think at this point running Python files feels like the best way to go (as opposed to YAML etc.).
Bonus: assuming we can pull off the conversion of all current yaml files to python, I think this would open up a really nice way of exposing these algo configs as proper examples in our docs. This section still bothers me.