Ross Wightman 31453b039e Update Auto/RandAugment comments, README, more. ... * Add a weighted choice option for RandAugment * Adjust magnitude noise/std naming, config		2019-11-22 13:24:52 -08:00
..
data	Update Auto/RandAugment comments, README, more.	2019-11-22 13:24:52 -08:00
loss	…
models	Add TF RandAug weights for B5/B7 EfficientNet models.	2019-10-30 16:49:17 -07:00
optim	Add Nvidia's NovogGrad impl from Jasper (cleaner/faster than current) and Apex Fused optimizers	2019-08-29 15:21:38 -07:00
scheduler	Add RAdam, NovoGrad, Lookahead, and AdamW optimizers, a few ResNet tweaks and scheduler factory tweak.	2019-08-28 00:14:10 -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	Add support for new AMP checkpointing support w/ amp.state_dict	2019-08-29 15:19:18 -07:00
version.py	Bump version for next push to pypi	2019-08-25 22:32:12 -07:00