61 Commits (update-test-pytorch)

Author	SHA1	Message	Date
Ross Wightman	40fea63ebe	Add checkpoint averaging script. Add headers, shebangs, exec perms to all scripts	5 years ago
Ross Wightman	506df0e3d0	Add CondConv support for EfficientNet into WIP for GenEfficientNet Feature extraction setup	5 years ago
Ross Wightman	c099374771	Map pretrained checkpoint to cpu to avoid issue with some pretrained checkpoints still having CUDA tensors. Fixes #42	5 years ago
Ross Wightman	b93fcf0708	Add Facebook Research Semi-Supervised and Semi-Weakly Supervised ResNet model weights.	5 years ago
Ross Wightman	aff194f42c	Merge pull request #32 from rwightman/opt ... More optimizer work	5 years ago
Ross Wightman	3d9c8a6489	Add support for new AMP checkpointing support w/ amp.state_dict	5 years ago
Ross Wightman	81875d52a6	Update sotabench model list, add Mean-Max pooling DPN variants, disable download progress	5 years ago
Minqin Chen	4e7a854dd0	Update helpers.py ... Fixing out of memory error by loading the checkpoint onto the CPU.	5 years ago
Ross Wightman	c6b32cbe73	A number of tweaks to arguments, epoch handling, config ... * reorganize train args * allow resolve_data_config to be used with dict args, not just arparse * stop incrementing epoch before save, more consistent naming vs csv, etc * update resume and start epoch handling to match above * stop auto-incrementing epoch in scheduler	6 years ago
Ross Wightman	6fc886acaf	Remove all prints, change most to logging calls, tweak alignment of batch logs, improve setup.py	6 years ago
Ross Wightman	aa4354f466	Big re-org, working towards making pip/module as 'timm'	6 years ago