Ross Wightman
|
9811e229f7
|
Fix regression in models with 1001 class pretrained weights. Improve batchnorm arg and BatchNormAct layer handling in several models.
|
2021-02-09 16:22:52 -08:00 |
|
Ross Wightman
|
231d04e91a
|
ResNetV2 pre-act and non-preact model, w/ BiT pretrained weights and support for ViT R50 model. Tweaks for in21k num_classes passing. More to do... tests failing.
|
2020-12-28 16:59:15 -08:00 |
|
Ross Wightman
|
b1f1a54de9
|
More uniform treatment of classifiers across all models, reduce code duplication.
|
2020-08-03 22:18:24 -07:00 |
|
Ross Wightman
|
4e61c6a12d
|
Cleanup, refactoring of Feature extraction code, add tests, fix tests, non hook feature extraction working with torchscript
|
2020-07-20 16:10:31 -07:00 |
|
Ross Wightman
|
9eba134d79
|
More models supporting feature extraction, xception, gluon_xception, inception_v3, inception_v4, pnasnet, nasnet, dla. Fix DLA unused projection params.
|
2020-07-19 14:02:02 -07:00 |
|
Ross Wightman
|
a66df5fb91
|
More model feature extraction support, start to deprecate senet.py, dilations added to regnet, add proper aligned xception
|
2020-07-03 00:41:30 -07:00 |
|
Ross Wightman
|
d0113f9cdb
|
Fix a few issues that came up in tests
|
2020-06-29 21:13:21 -07:00 |
|
Ross Wightman
|
d23a2697d0
|
Working on feature extraction, interfaces refined, a number of models working, some in progress.
|
2020-06-29 18:18:59 -07:00 |
|
Ross Wightman
|
0ea53cecc3
|
Merge branch 'master' into densenet_update_and_more
|
2020-05-22 16:18:10 -07:00 |
|
Ross Wightman
|
afb6bd0669
|
Add backward and default_cfg tests and fix a few issues found. Fix #153
|
2020-05-21 15:30:58 -07:00 |
Vyacheslav Shults
|
a7ebe09029
|
Replace all None by nn.Identity() in all models reset_classifier when False-values num_classes is given.
Make small code refactoring
|
2020-05-06 09:54:03 +03:00 |
|
Ross Wightman
|
13746a33fc
|
Big move, layer modules and fn to timm/models/layers
|
2020-02-09 13:13:08 -08:00 |
|
Ross Wightman
|
3bef524f9c
|
Finish with HRNet, weights and models updated. Improve consistency in model classifier/global pool treatment.
|
2019-11-29 17:56:36 -08:00 |
|
Ross Wightman
|
da52fcf78a
|
Add NASNet-Large model
|
2019-07-03 22:10:50 -07:00 |