mirrors/pytorch-image-models
1
0
Fork 0
mirror of https://github.com/huggingface/pytorch-image-models.git synced 2025-06-03 15:01:08 +08:00
Code Issues Projects Releases Wiki Activity
2,543 Commits 141 Branches 65 Tags
ff136b8d3a
Commit Graph

8 Commits

Author SHA1 Message Date
MengqingCao
234f975787 add npu support 2024-10-16 07:13:45 +00:00
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 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