Ross Wightman b1b6e7c361 Fix a few more issues related to #216 w/ TResNet (space2depth) and FP16 weights in wide resnets. Also don't completely dump pretrained weights in in_chans != 1 or 3 cases.		2020-08-11 18:57:47 -07: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	Fix a few more issues related to #216 w/ TResNet (space2depth) and FP16 weights in wide resnets. Also don't completely dump pretrained weights in in_chans != 1 or 3 cases.	2020-08-11 18:57:47 -07:00