Update Embed.py

EdwardTj · Nov 25, 2022 · 8798848 · 8798848
1 parent e1bfb66
commit 8798848
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/layers/Embed.py b/layers/Embed.py
@@ -28,7 +28,7 @@ def forward(self, x):
 class TokenEmbedding(nn.Module):
     def __init__(self, c_in, d_model):
         super(TokenEmbedding, self).__init__()
-        padding = 1 if torch.__version__ >= '1.5.0' else 2
+        padding = 1 if compared_version(torch.__version__, '1.5.0') else 2
         self.tokenConv = nn.Conv1d(in_channels=c_in, out_channels=d_model,
                                    kernel_size=3, padding=padding, padding_mode='circular', bias=False)
         for m in self.modules():