Ross Wightman 7ab2491ab7 Better handling of crossvit for tests / forward_features, fix torchscript regression in my changes

2021-09-13 13:01: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

Better handling of crossvit for tests / forward_features, fix torchscript regression in my changes

2021-09-13 13:01:05 -07:00

test_optim.py

Another attempt at sgd momentum test passing...

2021-09-03 20:50:26 -07:00