You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.

Ross Wightman 4e61c6a12d Cleanup, refactoring of Feature extraction code, add tests, fix tests, non hook feature extraction working with torchscript		4 years ago
..
__init__.py	Basic forward pass test for all registered models	4 years ago
test_layers.py	Add custom grad tests, fix cut & paste error with hard_mish ME, add a few more pytorch act fns to factory	4 years ago
test_models.py	Cleanup, refactoring of Feature extraction code, add tests, fix tests, non hook feature extraction working with torchscript	4 years ago