59 Commits (c976a410d9295224f23cf413466fd3387bf651c1)

Author	SHA1	Message	Date
Ross Wightman	4bb5e9b224	Ported Tensorflow pretrained EfficientNet weights and some model cleanup ... * B0-B3 weights ported from TF with close to paper accuracy * Renamed gen_mobilenet to gen_efficientnet since scaling params go well beyond 'mobile' specific * Add Tensorflow preprocessing option for closer images to source repo	6 years ago
Ross Wightman	db1fe34d0c	Update a few comment, add some references	6 years ago
Ross Wightman	0562b91c38	Add per model crop pct, interpolation defaults, tie it all together ... * create one resolve fn to pull together model defaults + cmd line args * update attribution comments in some models * test update train/validation/inference scripts	6 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	6 years ago
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	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	31055466fc	Fixup validate/inference script args, fix senet init for better test accuracy	6 years ago
Ross Wightman	5855b07ae0	Initial commit, puting some ol pieces together	6 years ago