Ross Wightman ef9c6fb846 forward_head(), consistent pre_logits handling to reduce likelihood of people manually replacing .head module having issues

2024-04-09 21:54:59 -07:00

..

data

…

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

forward_head(), consistent pre_logits handling to reduce likelihood of people manually replacing .head module having issues

2024-04-09 21:54:59 -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

Add support for dynamo based onnx export

2024-03-13 12:05:26 -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