Ross Wightman 678ba4e0a2 Add NFNet-F model weights ported from DeepMind Haiku impl and new set of models w/ compatible config.

2021-02-18 12:28:46 -08:00

..

data

Add new weights for ecaresnet26t/50t/269d models. Remove distinction between 't' and 'tn' (tiered models), tn is now t. Add test time img size spec to default cfg.

2021-02-06 16:30:02 -08:00

loss

…

models

Add NFNet-F model weights ported from DeepMind Haiku impl and new set of models w/ compatible config.

2021-02-18 12:28:46 -08:00

optim

…

scheduler

…

utils

Initial AGC impl. Still testing.

2021-02-15 23:22:44 -08:00

__init__.py

…

version.py

Update README for AGC and bump version to 0.4.4

2021-02-16 09:13:03 -08:00