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
|
3b9004bef9
|
Lots of changes to model creation helpers, close to finalizing feature extraction / interfaces
|
2020-07-17 17:54:26 -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 |
|
Ross Wightman
|
7df83258c9
|
Merge branch 'master' into densenet_update_and_more
|
2020-05-13 23:34:44 -07:00 |
|
Ross Wightman
|
17270c69b9
|
Remove annoying InceptionV3 dependency on scipy and insanely slow trunc_norm init. Bring InceptionV3 code into this codebase and use upcoming torch trunch_norm_ init.
|
2020-05-12 21:59:34 -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
|
d4debe6597
|
Update version, results csv files, and move remaining dropbox weights to github
|
2019-07-25 16:54:44 -07:00 |
|
Ross Wightman
|
171c0b88b6
|
Add model registry and model listing fns, refactor model_factory/create_model fn
|
2019-06-23 18:22:16 -07:00 |
|
Ross Wightman
|
6cc214bd7a
|
Consistency in model entrypoints
* move pretrained entrypoint arg to first pos to be closer to torchvision/hub
* change DPN weight URLS to my github location
|
2019-06-20 23:29:44 -07:00 |
|
Ross Wightman
|
aa4354f466
|
Big re-org, working towards making pip/module as 'timm'
|
2019-06-19 17:20:51 -07:00 |