Fix batch size calculation in dist_tuto

[petergtz]

Description

Batch size (bsz) must be an int, not a float. This change fixes it, basically doing the same as in https://github.com/seba-1511/dist_tuto.pth/blob/a552567061a9985cdcfe72ecb9b47e4630d6a7fe/train_dist.py#L85 (the sample linked from the tutorial).

Checklist

• [x] This is a trivial tutorial fix. I have not created an issue for it.
• [ ] The issue that is being fixed is referred in the description (see above "Fixes #ISSUE_NUMBER")
• [x] Only one issue is addressed in this pull request
• [ ] Labels from the issue that this PR is fixing are added to this pull request
• [x] No unnecessary issues are included into this pull request.

[pytorch-bot[bot]]

:link: Helpful Links

:test_tube: See artifacts and rendered test results at hud.pytorch.org/pr/pytorch/tutorials/2754

• :page_facing_up: Preview Python docs built from this PR

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

:heavy_exclamation_mark: 1 Active SEVs

There are 1 currently active SEVs. If your PR is affected, please view them below:

• Upgrade MacOS runner to 14

:white_check_mark: No Failures

As of commit 8d1c039f44732ec4c8ea030c03c1cbbb2d8505f1 with merge base cdbb559345a0513dc134a8d721f7d58928d9f6ff (): :green_heart: Looks good so far! There are no failures yet. :green_heart:

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