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	…