Ross Wightman
|
e57625e814
|
Tweak dist_backend to use device_type (before possible :)
|
2024-05-15 08:49:25 -07:00 |
Setepenre
|
8848dad362
|
Update distributed.py
|
2024-05-13 16:55:42 -04:00 |
|
Ross Wightman
|
47c9bc4dc6
|
Fix device idx split
|
2024-02-10 21:41:14 -08:00 |
|
Ross Wightman
|
a08b57e801
|
Fix distributed flag bug w/ flex device handling
|
2024-02-03 16:26:15 -08:00 |
|
Ross Wightman
|
a48ab818f5
|
Improving device flexibility in train. Fix #2081
|
2024-01-20 15:10:20 -08:00 |
|
Ross Wightman
|
87939e6fab
|
Refactor device handling in scripts, distributed init to be less 'cuda' centric. More device args passed through where needed.
|
2022-09-23 16:08:59 -07:00 |
|
Ross Wightman
|
532e3b417d
|
Reorg of utils into separate modules
|
2020-09-07 13:58:09 -07:00 |