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 bfc72f75d3
Expand scope of testing for non-std vision transformer / mlp models. Some related cleanup and create fn cleanup for all vision transformer and mlp models. More CoaT weights.
3 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 Expand scope of testing for non-std vision transformer / mlp models. Some related cleanup and create fn cleanup for all vision transformer and mlp models. More CoaT weights. 3 years ago