Ross Wightman 0dadb4a6e9 Initial multi-weight support, handled so old pretraing config handling co-exists with new tags.

2022-12-05 10:21:34 -08: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

Initial multi-weight support, handled so old pretraing config handling co-exists with new tags.

2022-12-05 10:21:34 -08:00

test_optim.py

Merge branch 'master' into attn_update

2021-09-13 16:51:10 -07:00

test_utils.py

Fix ordering of tests

2021-10-02 16:10:11 +01:00