pytorch-image-models

Commit Graph

Author	SHA1	Message	Date
Ross Wightman	1019414fd2	Add ported Tensorflow EfficientNet B4/B5 weights	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

Author

SHA1

Message

Date

Ross Wightman

1019414fd2

Add ported Tensorflow EfficientNet B4/B5 weights

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

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

3 Commits (1019414fd2b00d76b4c90dd5883691969be45996)