Ross Wightman 47c9bc4dc6 Fix device idx split

2024-02-10 21:41:14 -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

Add pos_weight and support for summing over classes to BCE impl in train scripts

2023-12-30 12:13:06 -08:00

models

Cleanup imports for vit relpos

2024-02-10 21:40:57 -08:00

optim

Update sgdw for older pytorch

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

scheduler

Allow training w/o validation split set

2024-01-08 09:38:42 -08:00

utils

Fix device idx split

2024-02-10 21:41:14 -08: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-01-08 09:39:22 -08:00