From 3f4448ccf0f17d1cc7ca4db5f7dc1babde953893 Mon Sep 17 00:00:00 2001 From: Vasilis Vryniotis Date: Tue, 31 May 2022 10:51:15 +0100 Subject: [PATCH 1/2] Add missing `_version` to the MLPBlock --- torchvision/models/vision_transformer.py | 1 + 1 file changed, 1 insertion(+) diff --git a/torchvision/models/vision_transformer.py b/torchvision/models/vision_transformer.py index 063d51749b4..5c2cf2b7466 100644 --- a/torchvision/models/vision_transformer.py +++ b/torchvision/models/vision_transformer.py @@ -39,6 +39,7 @@ class ConvStemConfig(NamedTuple): class MLPBlock(MLP): """Transformer MLP block.""" + _version = 2 def __init__(self, in_dim: int, mlp_dim: int, dropout: float): super().__init__(in_dim, [mlp_dim, in_dim], activation_layer=nn.GELU, inplace=None, dropout=dropout) From b9a18a95b95c7ceacf372202cf764b3622cd4105 Mon Sep 17 00:00:00 2001 From: Vasilis Vryniotis Date: Tue, 31 May 2022 10:56:05 +0100 Subject: [PATCH 2/2] fix linter --- torchvision/models/vision_transformer.py | 1 + 1 file changed, 1 insertion(+) diff --git a/torchvision/models/vision_transformer.py b/torchvision/models/vision_transformer.py index 5c2cf2b7466..e67c2a67acd 100644 --- a/torchvision/models/vision_transformer.py +++ b/torchvision/models/vision_transformer.py @@ -39,6 +39,7 @@ class ConvStemConfig(NamedTuple): class MLPBlock(MLP): """Transformer MLP block.""" + _version = 2 def __init__(self, in_dim: int, mlp_dim: int, dropout: float):