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.
pytorch-image-models/tests
Ross Wightman 8bf63b6c6c
Able to use other attn layer in EfficientNet now. Create test ECA + GC B0 configs. Make ECA more configurable.
4 years ago
..
__init__.py Basic forward pass test for all registered models 5 years ago
test_layers.py Fix #661, move hardswish out of default args for LeViT. Enable native torch support for hardswish, hardsigmoid, mish if present. 4 years ago
test_models.py Able to use other attn layer in EfficientNet now. Create test ECA + GC B0 configs. Make ECA more configurable. 4 years ago