pytorch-image-models/timm
Ross Wightman 7d4ada6d16 Update ViTamin model defs 2024-06-06 09:16:43 -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 ConvStem and MobileCLIP hybrid model for B variant. Add full norm disable support to ConvNormAct layers 2024-06-06 09:15:27 -07:00
loss
models Update ViTamin model defs 2024-06-06 09:16:43 -07:00
optim
scheduler
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
version.py Version 1.0.4.dev0 2024-05-20 15:52:27 -07:00