pytorch-image-models/timm
Ross Wightman 0737cf231d Add Next-ViT 2024-02-10 17:05:16 -08:00
..
data Small tweak of timm ToTensor for clarity 2024-02-10 14:57:40 -08:00
layers Make NormMlpClassifier head reset args consistent with ClassifierHead 2024-02-10 16:25:33 -08:00
loss
models Add Next-ViT 2024-02-10 17:05:16 -08:00
optim
scheduler
utils Fix distributed flag bug w/ flex device handling 2024-02-03 16:26:15 -08:00
__init__.py
version.py