tutorials icon indicating copy to clipboard operation
tutorials copied to clipboard

Published 20 hours ago • verified publisher icon pytorch

Fix batch size calculation in dist_tuto

Open petergtz opened this issue 6 months ago • 1 comments

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.

petergtz avatar Jan 29 '24 21:01 petergtz