pytorch-image-models

Commit Graph

Author	SHA1	Message	Date
Vyacheslav Shults	44ed0f85e8	Refactor imports and indents according to PEP. Add proposal for links to pretrained weights. Unify style of reset_classifier function. Replace None by nn.Identity() in torch modules (all empty classifiers and HRNet)	5 years ago
Ross Wightman	b8762cc67d	Model updates. Add my best ResNet50 weights top-1=78.47. Add some other torchvision weights. * Remove some models that don't exist as pretrained an likely never will (se)resnext152 * Add some torchvision weights as tv_ for models that I have added better weights for * Add wide resnet recently added to torchvision along with resnext101-32x8d * Add functionality to model registry to allow filtering on pretrained weight presence	6 years ago
Ross Wightman	171c0b88b6	Add model registry and model listing fns, refactor model_factory/create_model fn	6 years ago

Author

SHA1

Message

Date

Vyacheslav Shults

44ed0f85e8

Refactor imports and indents according to PEP.

Add proposal for links to pretrained weights.
Unify style of reset_classifier function.
Replace None by nn.Identity() in torch modules (all empty classifiers and HRNet)

Ross Wightman

b8762cc67d

Model updates. Add my best ResNet50 weights top-1=78.47. Add some other torchvision weights.

* Remove some models that don't exist as pretrained an likely never will (se)resnext152
* Add some torchvision weights as tv_ for models that I have added better weights for
* Add wide resnet recently added to torchvision along with resnext101-32x8d
* Add functionality to model registry to allow filtering on pretrained weight presence

Ross Wightman

171c0b88b6

Add model registry and model listing fns, refactor model_factory/create_model fn

3 Commits (a1886346df714b47880cc0f278c792f0a4c36261)