Skip to content

Comments

Add support for multiple gpus in training#53

Open
infiniterik wants to merge 1 commit intoShivanandroy:mainfrom
knoxml:main
Open

Add support for multiple gpus in training#53
infiniterik wants to merge 1 commit intoShivanandroy:mainfrom
knoxml:main

Conversation

@infiniterik
Copy link

  • Loggers updated to use sync_dist
  • Allow multiple gpus to be specified
  • Update pytorch-lightning to 2.0.1
  • specify ddp mode

- Loggers updated to use `sync_dist`
- Allow multiple gpus to be specified
- Update pytorch-lightning to 2.0.1
- specify ddp mode
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant