Ross Wightman
|
b1f1a54de9
|
More uniform treatment of classifiers across all models, reduce code duplication.
|
4 years ago |
Ross Wightman
|
6c17d57a2c
|
Fix some attributions, add copyrights to some file docstrings
|
5 years ago |
Ross Wightman
|
4e61c6a12d
|
Cleanup, refactoring of Feature extraction code, add tests, fix tests, non hook feature extraction working with torchscript
|
5 years ago |
Ross Wightman
|
9eba134d79
|
More models supporting feature extraction, xception, gluon_xception, inception_v3, inception_v4, pnasnet, nasnet, dla. Fix DLA unused projection params.
|
5 years ago |
Ross Wightman
|
d23a2697d0
|
Working on feature extraction, interfaces refined, a number of models working, some in progress.
|
5 years ago |
Ross Wightman
|
0ea53cecc3
|
Merge branch 'master' into densenet_update_and_more
|
5 years ago |
Ross Wightman
|
afb6bd0669
|
Add backward and default_cfg tests and fix a few issues found. Fix #153
|
5 years ago |
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
|
5 years ago |
Ross Wightman
|
13746a33fc
|
Big move, layer modules and fn to timm/models/layers
|
5 years ago |
Ross Wightman
|
3bef524f9c
|
Finish with HRNet, weights and models updated. Improve consistency in model classifier/global pool treatment.
|
5 years ago |
Ross Wightman
|
3b4868f6dc
|
A few more additions to Gluon Xception models to match interface of others.
|
5 years ago |
Ross Wightman
|
4d505e0785
|
Add working Gluon Xception-65 model. Some cleanup still needed.
|
5 years ago |