Ross Wightman c719f7eb86 More forward_intermediates() updates ...

* add convnext, resnet, efficientformer, levit support
* remove kwargs only for fn so that torchscript isn't broken for all :(
* use reset_classifier() consistently in prune

2024-05-03 16:22:32 -07:00

..

__init__.py

…

test_layers.py

device agnostic testing

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

test_models.py

More forward_intermediates() updates

2024-05-03 16:22:32 -07:00

test_optim.py

fix bug

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

test_utils.py

…