Ross Wightman a6af48be64 add madgradw optimizer		2021-08-17 22:19:27 -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	Add gc_efficientnetv2_rw_t weights (global context instead of SE attn). Add TF XL weights even though the fine-tuned ones don't validate that well. Change default arg for GlobalContext to use scal (mul) mode.	2021-08-07 16:45:29 -07:00
test_optim.py	add madgradw optimizer	2021-08-17 22:19:27 -07:00