Skip to content

Modernize the Ray Tune tutorial #3694

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.

Sign up for GitHub

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

Open
crypdick wants to merge 29 commits into
base: main
Choose a base branch
from
+405 −299
Open

Modernize the Ray Tune tutorial #3694

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
29 commits
Select commit Hold shift + click to select a range
6e33fc6
bump Ray version
Dec 9, 2025
bededb9
Ignore more data stuff
Dec 9, 2025
352ad9b
Update Ray Tune tutorial to use new API and improve formatting
Dec 9, 2025
10566e3
Remove Ray initialization from hyperparameter tuning tutorial and upd…
Dec 9, 2025
ad79672
Clean up hparam tuning tutorial, modernize checkpointing
Dec 9, 2025
a941761
Polish hparam tutorial a bit
Dec 10, 2025
64bc12e
Use the actual CIFAR10 normalization values
Dec 10, 2025
1f11769
polish
Dec 10, 2025
e360448
finalize the hyperparameter tuning tutorial
Dec 11, 2025
90de3a9
add author
crypdick Dec 12, 2025
52e6a68
Merge branch 'main' into ray-version-bump
svekars Dec 15, 2025
0ab1f75
Ignore more IDE stuff
Dec 15, 2025
ce0bc81
make linter happy
Dec 15, 2025
8c18e68
PR feedback
Dec 15, 2025
e1e0ea1
PR feedback
Dec 15, 2025
ffbb8cb
fix device loading
Dec 16, 2025
326952e
turn the pip code block to markdown section
Dec 16, 2025
a6b27be
pr feedback and linting
Dec 16, 2025
60d7e32
Restructure the hyperparameter tuning tutorial
Dec 16, 2025
025c44b
add GPUs back to default config
Dec 16, 2025
94faa53
Expose cpus_per_trial for configuration
Dec 16, 2025
d088014
Merge branch 'main' into ray-version-bump
svekars Jan 5, 2026
6ba2d6d
Update beginner_source/hyperparameter_tuning_tutorial.py
crypdick Jan 6, 2026
ecae2e9
switch CI machine to g5.4xlarge
Jan 6, 2026
947310f
revert num dataloader workers from 2 -> 8
Jan 6, 2026
1764df6
add cards after intro
Jan 6, 2026
46729ef
Merge branch 'main' into ray-version-bump
svekars Jan 6, 2026
a27085f
make linter happy
Jan 7, 2026
7dd199d
Apply suggestions from code review
crypdick Jan 7, 2026
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
2 changes: 1 addition & 1 deletion .ci/docker/requirements.txt
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -32,7 +32,7 @@ bs4
awscliv2==2.1.1
flask
spacy==3.4.1
ray[tune]==2.7.2
ray[tune]==2.52.1
tensorboard
jinja2==3.1.3
pytorch-lightning
Expand Down
7 changes: 6 additions & 1 deletion .gitignore
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -21,6 +21,8 @@ beginner_source/hymenoptera_data/
intermediate_source/data/
*.zip
MNIST/
data/cifar-10-batches-py/*
*.tar.gz

#builds
_build/
Expand Down Expand Up @@ -105,6 +107,7 @@ target/
celerybeat-schedule
# dotenv
.env
.envrc

# virtualenv
venv/
Expand All @@ -124,8 +127,10 @@ cleanup.sh
# PyTorch things
*.pt

# VSCode
# IDEs
*.vscode
.cfg/
.cursor

# pyspelling
dictionary.dic
Expand Down
3 changes: 2 additions & 1 deletion .jenkins/metadata.json
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -10,7 +10,8 @@
"duration": 1200
},
"beginner_source/hyperparameter_tuning_tutorial.py": {
"duration": 0
"needs": "linux.g5.4xlarge.nvidia.gpu",
"duration": 1800
},
"advanced_source/dynamic_quantization_tutorial.py": {
"duration": 380
Expand Down
Loading