4 Commits (99ab1b1276d8c35035eb5ea754296bc72c51d595)

Author	SHA1	Message	Date
Ross Wightman	89147a91e6	Remove adabound optimizer, never got it working well on larger datasets	2019-05-14 18:35:43 -07:00
Ross Wightman	bc264269c9	Morph mnasnet impl into a generic mobilenet that covers Mnasnet, MobileNetV1/V2, ChamNet, FBNet, and related ... * add an alternate RMSprop opt that applies eps like TF * add bn params for passing through alternates and changing defaults to TF style	2019-04-21 15:54:28 -07:00
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	2019-04-11 21:32:16 -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