Ross Wightman b41cffaa93 Fix a few issues loading pretrained vit/bit npz weights w/ num_classes=0 __init__ arg. Missed a few other small classifier handling detail on Mlp, GhostNet, Levit. Should fix #713		2021-06-22 23:16:05 -07:00
..
__init__.py	Basic forward pass test for all registered models	2020-05-07 00:20:58 -04:00
test_layers.py	Fix #661, move hardswish out of default args for LeViT. Enable native torch support for hardswish, hardsigmoid, mish if present.	2021-05-26 15:28:42 -07:00
test_models.py	Fix a few issues loading pretrained vit/bit npz weights w/ num_classes=0 __init__ arg. Missed a few other small classifier handling detail on Mlp, GhostNet, Levit. Should fix #713	2021-06-22 23:16:05 -07:00