TimmResnet26D

BloodAxe · Jan 21, 2024 · 6b9b717 · 6b9b717
1 parent d610f6c
commit 6b9b717
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/pytorch_toolbelt/modules/encoders/timm/resnet.py b/pytorch_toolbelt/modules/encoders/timm/resnet.py
@@ -169,7 +169,7 @@ def __init__(
         from timm.models.resnet import resnet26d
 
         act_layer = get_activation_block(activation)
-        encoder = resnet50d(features_only=True, pretrained=pretrained, act_layer=act_layer, **kwargs)
+        encoder = resnet26d(features_only=True, pretrained=pretrained, act_layer=act_layer, **kwargs)
         if first_conv_stride_one:
             encoder.conv1[0].stride = (1, 1)
             for info in encoder.feature_info: