14 Commits

Author	SHA1	Message	Date
Ross Wightman	58571e992e	Change block avgpool in senets to mean for performance issues with NVIDIA and AMP especially	2019-04-05 10:53:13 -07:00
Ross Wightman	6f9a0c8ef2	Merge branch 'master' of github.com:rwightman/pytorch-models	2019-04-01 11:07:05 -07:00
Ross Wightman	5cb1a35c6b	Fixup Resnext, remove alternate shortcut types	2019-04-01 11:03:37 -07:00
Ross Wightman	d87824bd65	Merge branch 'master' of github.com:rwightman/pytorch-models	2019-03-17 09:57:36 -07:00
Ross Wightman	45cde6f0c7	Improve creation of data pipeline with prefetch enabled vs disabled, fixup inception_res_v2 and dpn models	2019-03-11 22:17:42 -07:00
Ross Wightman	321435e6b4	Update resnext init	2019-03-10 14:24:53 -07:00
Ross Wightman	2295cf56c2	Add some Nvidia performance enhancements (prefetch loader, fast collate), and refactor some of training and model fact/transforms	2019-03-10 14:23:16 -07:00
Ross Wightman	9d927a389a	Add adabound, random erasing	2019-03-01 22:03:42 -08:00
Ross Wightman	1577c52976	Resnext added, changes to bring it and seresnet in line with rest of models	2019-03-01 15:44:04 -08:00
Ross Wightman	e0cfeb7d8e	Add some models, remove a model, tweak some models	2019-03-01 13:08:35 -08:00
Ross Wightman	31055466fc	Fixup validate/inference script args, fix senet init for better test accuracy	2019-02-22 14:07:50 -08:00
Ross Wightman	b5255960d9	Tweaking tanh scheduler, senet weight init (for BN), transform defaults	2019-02-13 23:11:09 -08:00
Ross Wightman	cf0c280e1b	Cleanup tranforms, add custom schedulers, tweak senet34 model	2019-02-06 20:19:11 -08:00
Ross Wightman	5855b07ae0	Initial commit, puting some ol pieces together	2019-02-01 22:07:34 -08:00