Skip to content

Commit

Permalink
[SDK] Use torchrun to create PyTorchJob from function (#2276)
Browse files Browse the repository at this point in the history 
* [SDK] Use torchrun to create PyTorchJob from function

Signed-off-by: Andrey Velichkevich <[email protected]>

* Update PyTorchJob SDK example

Signed-off-by: Andrey Velichkevich <[email protected]>

* Add consts for entrypoint

Signed-off-by: Andrey Velichkevich <[email protected]>

* Add check for num procs per worker

Signed-off-by: Andrey Velichkevich <[email protected]>

---------

Signed-off-by: Andrey Velichkevich <[email protected]>
  • Loading branch information
@andreyvelich
andreyvelich authored Oct 11, 2024
1 parent de6c28b commit 3541327
Show file tree
Hide file tree
Showing 5 changed files with 434 additions and 368 deletions.
Loading

0 comments on commit 3541327

Please sign in to comment.