Skip to content

[WIP] 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

Jump to bottom
Draft
crypdick wants to merge 10 commits into
base: main
Choose a base branch
from
Draft

[WIP] Modernize the Ray Tune tutorial #3694

crypdick wants to merge 10 commits into from

Conversation

@crypdick
Copy link

@crypdick crypdick commented Dec 11, 2025 •
edited by pytorch-bot bot
Loading

This is the first of several PRs against the Pytorch Tutorials. This PR unstales the Ray Tune tutorial:

  • bumps Ray version from 2.7.2 -> 2.52.1
  • update the tutorial with the latest code from the Pytorch CIFAR10 tutorial which it is based off of
  • update to use the latest Ray Tune APIs
  • fixed the norm values for CIFAR10
  • rewrote most of the prose
  • added new sections for sampling, observability, conclusion, additional links
  • updated tags in the doc indices. Will need approval on the new Ray-Distributed tag.

Opening this as a draft PR so that I can get feedback from the Ray Tune core devs first

cc @subramen @albanD @jbschlosser

@pytorch-bot
Copy link

pytorch-bot bot commented Dec 11, 2025 •
edited
Loading

🔗 Helpful Links

🧪 See artifacts and rendered test results at hud.pytorch.org/pr/pytorch/tutorials/3694

Note: Links to docs will display an error until the docs builds have been completed.

This comment was automatically generated by Dr. CI and updates every 15 minutes.

@meta-cla meta-cla bot added the cla signed label Dec 11, 2025
crypdick
crypdick commented Dec 11, 2025
View reviewed changes
beginner_source/hyperparameter_tuning_tutorial.py

# Set the number of trials, epochs, and GPUs per trial here:
# The following configuration is for a quick run (1 trial, 1 epoch, CPU only) for demonstration purposes.
main(num_trials=1, max_num_epochs=1, gpus_per_trial=0)
Copy link
Author

@crypdick crypdick Dec 11, 2025

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

TODO: add GPUs back when I am done

crypdick
crypdick commented Dec 11, 2025
View reviewed changes
beginner_source/hyperparameter_tuning_tutorial.py
def load_data(data_dir="./data"):
transform = transforms.Compose(
[transforms.ToTensor(), transforms.Normalize((0.5, 0.5, 0.5), (0.5, 0.5, 0.5))]
[transforms.ToTensor(), transforms.Normalize((0.4914, 0.48216, 0.44653), (0.2022, 0.19932, 0.20086))]
Copy link
Author

@crypdick crypdick Dec 11, 2025

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

I wrote a blog post about this: https://www.ricardodecal.com/guides/use-these-normalization-values-for-torchvision-datasets/

@svekars svekars added core Tutorials of any level of difficulty related to the core pytorch functionality intro labels Dec 11, 2025
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Reviewers

No reviews

Assignees

No one assigned

Labels

cla signed core Tutorials of any level of difficulty related to the core pytorch functionality intro

Projects

None yet

Milestone

No milestone

Development

Successfully merging this pull request may close these issues.

2 participants

@crypdick @svekars