FIX: Update optimization_tutorial.py to expose batch_size in train_loop()

[loganthomas]

Description

batch_size is not currently exposed as train_loop() parameter

• Re-implements FIX: Update optimization_tutorial.py to expose batch_size in train_loop() #2920 (accidental deletion from forked repo)

Checklist

• The issue that is being fixed is referred in the description (see above "Fixes #ISSUE_NUMBER")
• Only one issue is addressed in this pull request
• Labels from the issue that this PR is fixing are added to this pull request
• No unnecessary issues are included into this pull request.

cc @subramen @albanD @sekyondaMeta @svekars @kit1980 @brycebortree

[pytorch-bot]

🔗 Helpful Links

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

• 📄 Preview Python docs built from this PR

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

✅ No Failures

As of commit 73bfa34 with merge base 0740801 ():
💚 Looks good so far! There are no failures yet. 💚

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

[subramen]

Hi, thanks for the PR but the proposed change is not correct. The dataloader takes in the batch size arg (as seen here

tutorials/beginner_source/basics/optimization_tutorial.py

Line 47 in 0740801

train_dataloader = DataLoader(training_data, batch_size=64)

)

[loganthomas]

The dataloader takes in the batch size arg (as seen here)

Yes, but part of the train_loop uses the batch_size for reporting:

tutorials/beginner_source/basics/optimization_tutorial.py

Line 166 in 0740801

loss, current = loss.item(), batch * batch_size + len(X)

[github-actions]

Looks like this PR hasn't been updated in a while so we're going to go ahead and mark this as Stale.
Feel free to remove the Stale label if you feel this was a mistake.
If you are unable to remove the Stale label please contact a maintainer in order to do so.
If you want the bot to never mark this PR stale again, add the no-stale label.
Stale pull requests will automatically be closed after 30 days of inactivity.