pytorch-image-models

Commit Graph

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

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

2 Commits (7ba78aaaeb13d73ff20012d3d6c72beb031a4d26)