Ross Wightman cf5fec5047 Cleanup experimental vit weight init a bit

2021-03-20 09:44:24 -07:00

..

data

Improve evenness of per-worker split for validation set with TFDS

2021-03-18 23:16:14 -07:00

loss

asymmetric_loss

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

models

Cleanup experimental vit weight init a bit

2021-03-20 09:44:24 -07: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

Merge branch 'logger' into features. Change 'logger' to '_logger'.

2020-07-27 18:00:46 -07:00

utils

Initial AGC impl. Still testing.

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

__init__.py

Monster commit, activation refactor, VoVNet, norm_act improvements, more

2020-06-01 17:16:52 -07:00

version.py

Update README for AGC and bump version to 0.4.4

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