diff --git a/pytorch_toolbelt/modules/encoders/timm/resnet.py b/pytorch_toolbelt/modules/encoders/timm/resnet.py index ca7d387e3..b25922ac4 100644 --- a/pytorch_toolbelt/modules/encoders/timm/resnet.py +++ b/pytorch_toolbelt/modules/encoders/timm/resnet.py @@ -14,7 +14,7 @@ "TResNetMEncoder", "TimmResnet152D", "TimmSEResnet152D", - "TimmResnet26D" + "TimmResnet26D", "TimmResnet50D", "TimmResnet101D", "TimmResnet200D",