pytorch-image-models/timm
Ross Wightman cb33956b20 Fix some mistakes in mnv4 model defs 2024-05-23 14:24:32 -07:00
..
data
layers torchscript typing fixes 2024-05-23 11:43:05 -07:00
loss
models Fix some mistakes in mnv4 model defs 2024-05-23 14:24:32 -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