Skip to content
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.

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

Cosmetic changes to train.py #398

Merged
merged 5 commits into from
Jun 14, 2024
Merged

Conversation

kwen2501
Copy link
Contributor

@kwen2501 kwen2501 commented Jun 13, 2024

Stack from ghstack (oldest at bottom):

  • Move related code closer
  • Adjust comments

[ghstack-poisoned]
kwen2501 added a commit that referenced this pull request Jun 13, 2024
ghstack-source-id: 0a7e0ea2808c9cb22a9a939c7797f4d4cd745ffa
Pull Request resolved: #398
@facebook-github-bot facebook-github-bot added the CLA Signed This label is managed by the Meta Open Source bot. label Jun 13, 2024
Copy link
Contributor

@tianyu-l tianyu-l left a comment

Choose a reason for hiding this comment

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

beautiful!

train.py Outdated
@@ -142,15 +142,8 @@ def main(job_config: JobConfig):
torch.cuda.set_device(device)
init_distributed(job_config)

# Build meshes
Copy link
Contributor

Choose a reason for hiding this comment

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

very nit: Build -> build to keep style consistent

- Move related code closer
- Adjust comments

[ghstack-poisoned]
kwen2501 added a commit that referenced this pull request Jun 13, 2024
ghstack-source-id: eb73cf29319ed40d115f68655ee5e9ad4f7c02fe
Pull Request resolved: #398
- Move related code closer
- Adjust comments

[ghstack-poisoned]
kwen2501 added a commit that referenced this pull request Jun 13, 2024
ghstack-source-id: 75fe8612072535d205993416cfeca911099ea40d
Pull Request resolved: #398
- Move related code closer
- Adjust comments

[ghstack-poisoned]
kwen2501 added a commit that referenced this pull request Jun 13, 2024
ghstack-source-id: 7cb864d3bee58b3a65aff62c60f871712f83b242
Pull Request resolved: #398
- Move related code closer
- Adjust comments

[ghstack-poisoned]
kwen2501 added a commit that referenced this pull request Jun 14, 2024
ghstack-source-id: ce4a5b0b6b785ce595487c9d565a8af030c9d07b
Pull Request resolved: #398
@kwen2501 kwen2501 merged commit 36ae0fe into gh/kwen2501/5/base Jun 14, 2024
5 checks passed
kwen2501 added a commit that referenced this pull request Jun 14, 2024
ghstack-source-id: ce4a5b0b6b785ce595487c9d565a8af030c9d07b
Pull Request resolved: #398
@kwen2501 kwen2501 deleted the gh/kwen2501/5/head branch June 14, 2024 06:50
tianyu-l pushed a commit to tianyu-l/torchtitan_intern24 that referenced this pull request Aug 16, 2024
ghstack-source-id: ce4a5b0b6b785ce595487c9d565a8af030c9d07b
Pull Request resolved: pytorch#398
philippguevorguian pushed a commit to YerevaNN/YNNtitan that referenced this pull request Aug 17, 2024
ghstack-source-id: ce4a5b0b6b785ce595487c9d565a8af030c9d07b
Pull Request resolved: pytorch#398
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
CLA Signed This label is managed by the Meta Open Source bot.
Projects
None yet
Development

Successfully merging this pull request may close these issues.

3 participants