Ross Wightman 679daef76a More forward_intermediates() & features_only work ...

* forward_intermediates() added to beit, deit, eva, mvitv2, twins, vit, vit_sam
* add features_only to forward intermediates to allow just intermediate features
* fix #2060
* fix #1374
* fix #657

2024-04-09 21:29:16 -07: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

More forward_intermediates() & features_only work

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

optim

…

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

…

version.py

Update version.py

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