8 Commits

Author	SHA1	Message	Date
Ross Wightman	0bc50e84f8	Lots of refactoring and cleanup. ... * Move 'test time pool' to Module that can be used by any model, remove from DPN * Remove ResNext model file and combine with ResNet * Remove fbresnet200 as it was an old conversion and pretrained performance not worth param count * Cleanup adaptive avgmax pooling and add back conctat variant * Factor out checkpoint load fn	2019-04-10 14:53:34 -07:00
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	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