Commit Graph

10 Commits (e2cc4813105734f3ca41c94226f07e3f6e24b8e1)

Author SHA1 Message Date
Ross Wightman d23a2697d0 Working on feature extraction, interfaces refined, a number of models working, some in progress.
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
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 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 171c0b88b6 Add model registry and model listing fns, refactor model_factory/create_model fn
5 years ago
Ross Wightman 87b92c528e Some pretrianed URL changes
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