Ross Wightman 317ea3e599 Tweak NFNet, ResNetV2, RexNet feature extraction, use pre-act features for simplicty. Update tests for pytorch 1.8. 0.4.5 version bump.		2021-03-06 12:48:06 -08:00
..
__init__.py	Basic forward pass test for all registered models	2020-05-07 00:20:58 -04:00
test_layers.py	Add custom grad tests, fix cut & paste error with hard_mish ME, add a few more pytorch act fns to factory	2020-06-11 14:49:23 -07:00
test_models.py	Tweak NFNet, ResNetV2, RexNet feature extraction, use pre-act features for simplicty. Update tests for pytorch 1.8. 0.4.5 version bump.	2021-03-06 12:48:06 -08:00