Commit Graph

17 Commits (8a33a6c90a2b74c50f0129926f1830e5469e5f74)

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