Merge branch 'develop' into feat/pairlistconfig

2023-05-21 09:54:17 +02:00
parent 818a3342b9 a87b215d67
commit bf9a6dd6e7
1 changed files with 2 additions and 2 deletions
@@ -1,7 +1,7 @@
 import math
 import torch
-import torch.nn as nn
+from torch import nn
 """
@@ -68,7 +68,7 @@ class PyTorchTransformerModel(nn.Module):
        return x
-class PositionalEncoding(torch.nn.Module):
+class PositionalEncoding(nn.Module):
    def __init__(self, d_model, max_len=5000):
        """
        Args