Ross Wightman 0706d05d52 Benchmark models listed in txt file. Add more hybrid vit variants for testing

2021-02-28 16:00:33 -08:00

..

data

Add epoch-repeats arg to multiply the number of dataset passes per epoch. Currently for iterable datasets (read TFDS wrapper) only.

2021-02-23 17:31:42 -08:00

loss

asymmetric_loss

2020-10-16 17:12:28 +03:00

models

Benchmark models listed in txt file. Add more hybrid vit variants for testing

2021-02-28 16:00:33 -08:00

optim

Add benchmark.py script, and update optimizer factory to be more friendly to use outside of argparse interface.

2021-02-23 15:38:12 -08:00

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