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	0004f37d25	Some TResNet cleanup. * allow use of global pool arg, test-time-pooling * clean checkpoints to just contain state dict, add 448 res checkpoints * support DataParallel via lazy filter creation for JIT Downsample * some minor formatting (mostly alignment) preferences	5 years ago
Ross Wightman	13746a33fc	Big move, layer modules and fn to timm/models/layers	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

0004f37d25

Some TResNet cleanup.

* allow use of global pool arg, test-time-pooling
* clean checkpoints to just contain state dict, add 448 res checkpoints
* support DataParallel via lazy filter creation for JIT Downsample
* some minor formatting (mostly alignment) preferences

Ross Wightman

13746a33fc

Big move, layer modules and fn to timm/models/layers

3 Commits (2e085f9c6b2a38432cdd4edfaf6a3a5a64aefff4)