Ross Wightman 24f6d4f7f8 Fix #2127 move to ema device

2024-04-10 21:29:09 -07:00

..

data

Add teddy-bear class back to first 1000 classes of imagenet22k_ms_synsets (index 851)

2024-04-09 09:33:08 -07:00

layers

Make NormMlpClassifier head reset args consistent with ClassifierHead

2024-02-10 16:25:33 -08:00

loss

Fix #2132, remove use of _C.set_grad_enable. Line endings were messed up too

2024-04-09 09:00:23 -07:00

models

Fix #2139, disable strict weight loading when head changes from classification

2024-04-09 08:41:37 -07:00

optim

Update sgdw for older pytorch

2023-12-11 12:10:29 -08:00

scheduler

more

2024-04-02 10:25:30 +08:00

utils

Fix #2127 move to ema device

2024-04-10 21:29:09 -07:00

__init__.py

Major module / path restructure, timm.models.layers -> timm.layers, add _ prefix to all non model modules in timm.models

2022-12-06 15:00:06 -08:00

version.py

Update version.py

2024-02-19 11:08:17 -08:00