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
Ross Wightman	2295cf56c2	Add some Nvidia performance enhancements (prefetch loader, fast collate), and refactor some of training and model fact/transforms	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

Ross Wightman

2295cf56c2

Add some Nvidia performance enhancements (prefetch loader, fast collate), and refactor some of training and model fact/transforms

3 Commits (51f85702a76984498323c3d81f2e7df7047c3a99)