Ross Wightman a5a2ad2e48 Fix consistency, testing for forward_head w/ pre_logits, reset_classifier, models with pre_logits size != unpooled feature size ...

* add test that model supports forward_head(x, pre_logits=True)
* add head_hidden_size attr to all models and set differently from num_features attr when head has hidden layers
* test forward_features() feat dim == model.num_features and pre_logits feat dim == self.head_hidden_size
* more consistency in reset_classifier signature, add typing
* asserts in some heads where pooling cannot be disabled
Fix #2194

2024-06-07 13:53:00 -07:00

..

__init__.py

…

test_layers.py

device agnostic testing

2023-11-16 20:27:59 -08:00

test_models.py

Fix consistency, testing for forward_head w/ pre_logits, reset_classifier, models with pre_logits size != unpooled feature size

2024-06-07 13:53:00 -07:00

test_optim.py

fix bug

2023-11-16 20:27:59 -08:00

test_utils.py

…