pytorch-image-models/timm
Ross Wightman 70176a2dae torchscript typing fixes 2024-05-23 11:43:05 -07:00
..
data
layers torchscript typing fixes 2024-05-23 11:43:05 -07:00
loss
models Merge remote-tracking branch 'origin/main' into efficientnet_x 2024-05-23 11:01:39 -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