Commit Graph

19 Commits (bcec14d3b585d7b5c469705f99fb2d830bdcdb7d)

Author SHA1 Message Date
Ross Wightman d584e7f617 Support for huggingface hub via create_model and default_cfgs.
3 years ago
Ross Wightman b1f1a54de9 More uniform treatment of classifiers across all models, reduce code duplication.
4 years ago
Ross Wightman 3b9004bef9 Lots of changes to model creation helpers, close to finalizing feature extraction / interfaces
4 years ago
Ross Wightman d23a2697d0 Working on feature extraction, interfaces refined, a number of models working, some in progress.
4 years ago
Ross Wightman a7e8cadd15 Remove pointless densenet configs, add an iabn version of 264 as it makes more sense to try someday...
4 years ago
Ross Wightman e78daf586a better densenet121 and densenetblur121d weights
4 years ago
Ross Wightman eb7653614f Monster commit, activation refactor, VoVNet, norm_act improvements, more
4 years ago
Ross Wightman 6441e9cc1b Fix memory_efficient mode for DenseNets. Add AntiAliasing (Blur) support for DenseNets and create one test model. Add lr cycle/mul params to train args.
4 years ago
Ross Wightman 780860d140 Add norm_act factory method, move JIT of norm layers to factory
4 years ago
Ross Wightman 14edacdf9a DenseNet converted to support ABN (norm + act) modules. Experimenting with EvoNorm, IABN
4 years ago
Ross Wightman 022ed001f3 Update DenseNet to latest in Torchvision (torchscript compat, checkpointing, proper init). Start adding ehanced configurability, stem options...
4 years ago
Vyacheslav Shults a7ebe09029 Replace all None by nn.Identity() in all models reset_classifier when False-values num_classes is given.
4 years ago
Ross Wightman 13746a33fc Big move, layer modules and fn to timm/models/layers
4 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 949b7a81c4 Fix typo in Densenet default resolutions
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 6fc886acaf Remove all prints, change most to logging calls, tweak alignment of batch logs, improve setup.py
5 years ago
Ross Wightman aa4354f466 Big re-org, working towards making pip/module as 'timm'
5 years ago