66 Commits (cd23f553974f0cda6b07761ba2d4a88e82236966)

Author	SHA1	Message	Date
Ross Wightman	c328b155e9	Random erasing crash fix and args pass through	5 years ago
Ross Wightman	9c3859fb9c	Uniform pretrained model handling. ... * All models have 'default_cfgs' dict * load/resume/pretrained helpers factored out * pretrained load operates on state_dict based on default_cfg * test all models in validate * schedule, optim factor factored out * test time pool wrapper applied based on default_cfg	5 years ago
Ross Wightman	f1cd1a5ce3	Cleanup CheckpointSaver, add support for increasing or decreasing metric, switch to prec1 metric in train loop	5 years ago
Ross Wightman	5180f94c7e	Distributed (multi-process) train, multi-gpu single process train, and NVIDIA AMP support	5 years ago
Ross Wightman	45cde6f0c7	Improve creation of data pipeline with prefetch enabled vs disabled, fixup inception_res_v2 and dpn models	5 years ago
Ross Wightman	2295cf56c2	Add some Nvidia performance enhancements (prefetch loader, fast collate), and refactor some of training and model fact/transforms	5 years ago
Ross Wightman	9d927a389a	Add adabound, random erasing	5 years ago
Ross Wightman	1577c52976	Resnext added, changes to bring it and seresnet in line with rest of models	5 years ago
Ross Wightman	31055466fc	Fixup validate/inference script args, fix senet init for better test accuracy	5 years ago
Ross Wightman	b1a5a71151	Update schedulers	5 years ago
Ross Wightman	b5255960d9	Tweaking tanh scheduler, senet weight init (for BN), transform defaults	5 years ago
Ross Wightman	a336e5bff3	Minor updates	5 years ago
Ross Wightman	cf0c280e1b	Cleanup tranforms, add custom schedulers, tweak senet34 model	5 years ago
Ross Wightman	c57717d325	Fix tta train bug, improve logging	5 years ago
Ross Wightman	72b4d162a2	Increase training performance	5 years ago
Ross Wightman	5855b07ae0	Initial commit, puting some ol pieces together	5 years ago