Ross Wightman 0ea53cecc3 Merge branch 'master' into densenet_update_and_more		2020-05-22 16:18:10 -07:00
..
data	added MultiEpochsDataLoader	2020-05-05 14:47:02 +03:00
loss	Update AugMix, JSD, etc comments and references	2020-01-11 12:02:05 -08:00
models	Merge branch 'master' into densenet_update_and_more	2020-05-22 16:18:10 -07:00
optim	Add 'momentum' optimizer (SGD w/o nesterov) for stable EfficientDet training defaults	2020-04-25 19:42:13 -07:00
scheduler	Fix memory_efficient mode for DenseNets. Add AntiAliasing (Blur) support for DenseNets and create one test model. Add lr cycle/mul params to train args.	2020-05-22 16:16:45 -07:00
__init__.py	Add model registry and model listing fns, refactor model_factory/create_model fn	2019-06-23 18:22:16 -07:00
utils.py	Remove poorly named metrics from torch imagenet example origins. Use top1/top5 in csv output for consistency with existing validation results files, acc elsewhere. Fixes #111	2020-04-10 14:41:08 -07:00
version.py	Fix #139. Broken SKResNets after BlurPool addition, as a plus, SKResNets support AA now too.	2020-05-04 12:14:43 -07:00