Ross Wightman 392595c7eb Add pool_size to default cfgs for new models to prevent tests from failing. Add explicit 200D_320 model entrypoint for next benchmark run.

2020-12-18 21:28:47 -08:00

..

data

Fix a bug with accuracy retrieving from RealLabels

2020-12-04 16:12:50 +01:00

loss

…

models

Add pool_size to default cfgs for new models to prevent tests from failing. Add explicit 200D_320 model entrypoint for next benchmark run.

2020-12-18 21:28:47 -08:00

optim

Configure create_optimizer with args.opt_args

2020-12-08 00:03:09 -06:00

scheduler

…

utils

Add separate set and update method to ModelEmaV2

2020-12-03 10:05:09 -08:00

__init__.py

…

version.py

Version 0.3.2

2020-11-30 13:39:08 -08:00