Commit Graph

21 Commits (81ca323751505c7e4ee9b7561020fd3c6008d116)

Author SHA1 Message Date
Fredo Guan 81ca323751
Davit update formatting and fix grad checkpointing (#7)
1 year ago
Ross Wightman 927f031293 Major module / path restructure, timm.models.layers -> timm.layers, add _ prefix to all non model modules in timm.models
1 year ago
Ross Wightman 372ad5fa0d Significant model refactor and additions:
2 years ago
Ross Wightman abc9ba2544 Transitioning default_cfg -> pretrained_cfg. Improving handling of pretrained_cfg source (HF-Hub, files, timm config, etc). Checkpoint handling tweaks.
2 years ago
Ross Wightman d5473c17f7 Fix incorrect name of shortcut/identity paths in many residual nets. Inherited from naming in old old torchvision, long fixed there.
3 years ago
Ross Wightman d584e7f617 Support for huggingface hub via create_model and default_cfgs.
3 years ago
Ross Wightman d72ddafe56 Fix some checkpoint / model str regressions
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 4ddde1d3a4 Fix two regressions
4 years ago
Ross Wightman eb7653614f Monster commit, activation refactor, VoVNet, norm_act improvements, more
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 ea300709f0 Fix model create fn not passing num_classes through. Fix #135
4 years ago
Ross Wightman f902bcd54c Layer refactoring continues, ResNet downsample rewrite for proper dilation in 3x3 and avg_pool cases
4 years ago
Ross Wightman 13746a33fc Big move, layer modules and fn to timm/models/layers
4 years ago
Ross Wightman a9d2424fd1 Add separate zero_init_last_bn function to support more block variety without a mess
4 years ago
Ross Wightman cefc9b7761 Move SelectKernelConv to conv2d_layers and more
4 years ago
Ross Wightman 53001dd292 ResNet / Res2Net additions:
4 years ago
rwightman d3ba34ee7e Fix Mobilenet V3 model name for sotabench. Minor res2net cleanup.
5 years ago
Ross Wightman 2680ad14bb Add Res2Net and DLA to README
5 years ago
rwightman adbf770f16 Add Res2Net and DLA models w/ pretrained weights. Update sotabench.
5 years ago