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
|
f31933cb37
|
Initial Vision Transformer impl w/ patch and hybrid variants. Refactor tuple helpers.
|
2020-10-13 13:33:44 -07:00 |
Yusuke Uchida
|
f6b56602f9
|
fix test_model_default_cfgs
|
2020-08-11 23:23:57 +09:00 |
Ross Wightman
|
6c17d57a2c
|
Fix some attributions, add copyrights to some file docstrings
|
2020-07-27 13:44:56 -07:00 |
Ross Wightman
|
08016e839d
|
Cleanup FeatureInfo getters, add TF models sourced Xception41/65/71 weights
|
2020-07-24 17:59:21 -07:00 |
Ross Wightman
|
2ac663f340
|
Add feature support to legacy senets, add 32x32 resnext models to exclude list for feature testing.
|
2020-07-21 11:15:30 -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
|
3b9004bef9
|
Lots of changes to model creation helpers, close to finalizing feature extraction / interfaces
|
2020-07-17 17:54:26 -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 |