pytorch-image-models/timm
Ross Wightman 015fbe457a Merge branch 'MengqingCao-npu_support' into device_amp_cleanup 2024-10-18 14:50:44 -07:00
..
data fix device check 2024-10-17 12:38:02 +00:00
layers Cleanup some amp related behaviour to better support different (non-cuda) devices 2024-10-18 13:54:16 -07:00
loss
models mambaout.py: fixed bug 2024-10-17 01:03:28 +03:30
optim
scheduler
utils Merge branch 'MengqingCao-npu_support' into device_amp_cleanup 2024-10-18 14:50:44 -07:00
__init__.py
version.py Version 1.0.11 2024-10-16 14:14:44 -07:00