Ross Wightman 59239d9df5 Cleanup imports for vit relpos		2024-02-10 21:40:57 -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	Cleanup imports for vit relpos	2024-02-10 21:40:57 -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	…