pytorch-image-models

Commit Graph

Author	SHA1	Message	Date
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	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

Author

SHA1

Message

Date

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

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

2 Commits (419555be62d1e5d023248c24c6232532d601887b)