Ross Wightman c099374771 Map pretrained checkpoint to cpu to avoid issue with some pretrained checkpoints still having CUDA tensors. Fixes #42		2019-10-19 17:27:46 -07:00
..
data	Fix AutoAugment abs translate calc	2019-09-01 22:07:45 -07:00
loss	Big re-org, working towards making pip/module as 'timm'	2019-06-19 17:20:51 -07:00
models	Map pretrained checkpoint to cpu to avoid issue with some pretrained checkpoints still having CUDA tensors. Fixes #42	2019-10-19 17:27:46 -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