Ross Wightman
|
56b90317cd
|
Change torchrun args to use _ instead of -, - is the new format, but looks like _ still works for backward compat with old versions. Fix #1742
|
2023-03-26 20:23:55 -07:00 |
Ross Wightman
|
2054f11c6f
|
Switch to torchrun as old launcher changed arg names in 2.0.. Fix #1728
|
2023-03-20 08:15:40 -07:00 |
Csaba Kertesz
|
5114c214fc
|
Change the Python interpreter to Python 3.x in the scripts
|
2021-02-09 21:20:28 +02:00 |
Ross Wightman
|
b0158a593e
|
Fix distributed train script
|
2019-04-05 20:49:58 -07:00 |
Ross Wightman
|
5180f94c7e
|
Distributed (multi-process) train, multi-gpu single process train, and NVIDIA AMP support
|
2019-04-05 10:53:04 -07:00 |