Ross Wightman 5fa6efa158 Add anti-aliasing support to mobilenetv3 and efficientnet family models. Update MobileNetV4 model defs, resolutions. Fix #599 ...

* create_aa helper function centralized for all timm uses (resnet, convbnact helper)
* allow BlurPool w/ pre-defined channels (expand)
* mobilenetv4 UIB block using ConvNormAct layers for improved clarity, esp with AA added
* improve more mobilenetv3 and efficientnet related type annotations

2024-05-27 22:06:22 -07:00

..

data

Add normalize flag to transforms factory, allow return of non-normalized native dtype torch.Tensors

2024-05-13 15:23:25 -07:00

layers

Add anti-aliasing support to mobilenetv3 and efficientnet family models. Update MobileNetV4 model defs, resolutions. Fix #599

2024-05-27 22:06:22 -07:00

loss

Fix #2132, remove use of _C.set_grad_enable. Line endings were messed up too

2024-04-09 09:00:23 -07:00

models

Add anti-aliasing support to mobilenetv3 and efficientnet family models. Update MobileNetV4 model defs, resolutions. Fix #599

2024-05-27 22:06:22 -07:00

optim

Update sgdw for older pytorch

2023-12-11 12:10:29 -08:00

scheduler

more

2024-04-02 10:25:30 +08:00

utils

Add regex matching support to AttentionExtract. Add return_dict support to graph extractors and use returned output in AttentionExtractor

2024-05-22 14:33:39 -07:00

__init__.py

Major module / path restructure, timm.models.layers -> timm.layers, add _ prefix to all non model modules in timm.models

2022-12-06 15:00:06 -08:00

version.py

Version 1.0.4.dev0

2024-05-20 15:52:27 -07:00