Revert changes of embedding

Author: hyunwoongko

.idea/.gitignore

@@ -26,10 +26,10 @@ def __init__(self, d_model, ffn_hidden, n_head, drop_prob):
         self.norm3 = LayerNorm(d_model=d_model)
         self.dropout3 = nn.Dropout(p=drop_prob)
 
-    def forward(self, dec, enc, t_mask, s_mask):
+    def forward(self, dec, enc, trg_mask, src_mask):
         # 1. compute self attention
         _x = dec
-        x = self.self_attention(q=dec, k=dec, v=dec, mask=t_mask)
+        x = self.self_attention(q=dec, k=dec, v=dec, mask=trg_mask)
 
         # 2. add and norm
         x = self.dropout1(x)
@@ -38,7 +38,7 @@ def forward(self, dec, enc, t_mask, s_mask):
         if enc is not None:
             # 3. compute encoder - decoder attention
             _x = x
-            x = self.enc_dec_attention(q=x, k=enc, v=enc, mask=s_mask)
+            x = self.enc_dec_attention(q=x, k=enc, v=enc, mask=src_mask)
 
             # 4. add and norm
             x = self.dropout2(x)

.idea/inspectionProfiles/profiles_settings.xml

@@ -22,10 +22,10 @@ def __init__(self, d_model, ffn_hidden, n_head, drop_prob):
         self.norm2 = LayerNorm(d_model=d_model)
         self.dropout2 = nn.Dropout(p=drop_prob)
 
-    def forward(self, x, s_mask):
+    def forward(self, x, src_mask):
         # 1. compute self attention
         _x = x
-        x = self.attention(q=x, k=x, v=x, mask=s_mask)
+        x = self.attention(q=x, k=x, v=x, mask=src_mask)
 
         # 2. add and norm
         x = self.dropout1(x)

.idea/misc.xml

@@ -25,10 +25,10 @@ def __init__(self, enc_voc_size, max_len, d_model, ffn_hidden, n_head, n_layers,
                                                   drop_prob=drop_prob)
                                      for _ in range(n_layers)])
 
-    def forward(self, x, s_mask):
+    def forward(self, x, src_mask):
         x = self.emb(x)
 
         for layer in self.layers:
-            x = layer(x, s_mask)
+            x = layer(x, src_mask)
 
         return x