Ross Wightman 5f12de4875 Add initial AttentionPool2d that's being trialed. Fix comment and still trying to improve reliability of sgd test.

2021-09-05 12:41:14 -07:00

..

__init__.py

…

test_layers.py

…

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 initial AttentionPool2d that's being trialed. Fix comment and still trying to improve reliability of sgd test.

2021-09-05 12:41:14 -07:00