From 33c6bdcb60d76d4127d038550585348a88a2ba89 Mon Sep 17 00:00:00 2001 From: peterkulits <5331526+kulits@users.noreply.github.com> Date: Wed, 4 Jan 2023 13:20:04 -0700 Subject: [PATCH] Update timm model name `vit_base_resnet50_384` was a deprecated (now-removed) alias for `vit_base_r50_s16_384` (see [[here](https://github.com/rwightman/pytorch-image-models/blob/v0.4.12/timm/models/vision_transformer_hybrid.py)]). This PR makes `omnidata-tools` compatible with recent versions of `timm` and does not affect compatibility with the `timm` version listed in the existing `requirements.txt`. --- omnidata_tools/torch/modules/midas/vit.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/omnidata_tools/torch/modules/midas/vit.py b/omnidata_tools/torch/modules/midas/vit.py index 80ba7ad..fe0e4a7 100644 --- a/omnidata_tools/torch/modules/midas/vit.py +++ b/omnidata_tools/torch/modules/midas/vit.py @@ -480,7 +480,7 @@ def _make_vit_b_rn50_backbone( def _make_pretrained_vitb_rn50_384( pretrained, use_readout="ignore", hooks=None, use_vit_only=False ): - model = timm.create_model("vit_base_resnet50_384", pretrained=pretrained) + model = timm.create_model("vit_base_r50_s16_384", pretrained=pretrained) hooks = [0, 1, 8, 11] if hooks == None else hooks return _make_vit_b_rn50_backbone(