Commit Graph

14 Commits (fc8b8afb6f0144afcbd927eb25546864e8a283b4)

Author SHA1 Message Date
Yusuke Uchida f6b56602f9 fix test_model_default_cfgs
4 years ago
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
4 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.
4 years ago
Ross Wightman 3b9004bef9 Lots of changes to model creation helpers, close to finalizing feature extraction / interfaces
4 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
Ross Wightman 7df83258c9 Merge branch 'master' into densenet_update_and_more
5 years ago
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.
5 years ago
Vyacheslav Shults a7ebe09029 Replace all None by nn.Identity() in all models reset_classifier when False-values num_classes is given.
5 years ago
Ross Wightman d4debe6597 Update version, results csv files, and move remaining dropbox weights to github
5 years ago
Ross Wightman 171c0b88b6 Add model registry and model listing fns, refactor model_factory/create_model fn
5 years ago
Ross Wightman 6cc214bd7a Consistency in model entrypoints
5 years ago
Ross Wightman aa4354f466 Big re-org, working towards making pip/module as 'timm'
5 years ago