55 Commits (21812d33aa60b9a2dde85968a8a5031424f9404e)

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	5180f94c7e	Distributed (multi-process) train, multi-gpu single process train, and NVIDIA AMP support	2019-04-05 10:53:04 -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	31055466fc	Fixup validate/inference script args, fix senet init for better test accuracy	2019-02-22 14:07:50 -08:00
Ross Wightman	5855b07ae0	Initial commit, puting some ol pieces together	2019-02-01 22:07:34 -08:00