Ross Wightman ecc7552c5c Add levit, levit_c, and visformer model defs. Largely untested and not finished cleanup.

2021-05-14 17:16:34 -07:00

..

__init__.py

Basic forward pass test for all registered models

2020-05-07 00:20:58 -04:00

test_layers.py

Add custom grad tests, fix cut & paste error with hard_mish ME, add a few more pytorch act fns to factory

2020-06-11 14:49:23 -07:00

test_models.py

Add levit, levit_c, and visformer model defs. Largely untested and not finished cleanup.

2021-05-14 17:16:34 -07:00