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	2019-04-11 21:32:16 -07:00
Ross Wightman	0bc50e84f8	Lots of refactoring and cleanup. * Move 'test time pool' to Module that can be used by any model, remove from DPN * Remove ResNext model file and combine with ResNet * Remove fbresnet200 as it was an old conversion and pretrained performance not worth param count * Cleanup adaptive avgmax pooling and add back conctat variant * Factor out checkpoint load fn	2019-04-10 14:53:34 -07:00
Ross Wightman	e0cfeb7d8e	Add some models, remove a model, tweak some models	2019-03-01 13:08:35 -08:00

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

2019-04-11 21:32:16 -07:00

Ross Wightman

0bc50e84f8

Lots of refactoring and cleanup.

* Move 'test time pool' to Module that can be used by any model, remove from DPN
* Remove ResNext model file and combine with ResNet
* Remove fbresnet200 as it was an old conversion and pretrained performance not worth param count
* Cleanup adaptive avgmax pooling and add back conctat variant
* Factor out checkpoint load fn

2019-04-10 14:53:34 -07:00

Ross Wightman

e0cfeb7d8e

Add some models, remove a model, tweak some models

2019-03-01 13:08:35 -08:00

3 Commits (c328b155e93dc637efd484cc43f4c5f3353a167e)