Skip to content

Commit d64c105

Browse files
Update by commit
1 parent d6248c7 commit d64c105

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

Diff for: embedding.py

+2-2
Original file line numberDiff line numberDiff line change
@@ -148,7 +148,7 @@ def __init__(self, vocab_size, embed_size = 512):
148148

149149
# architecture
150150
self.embed = nn.Embedding(vocab_size, dim, padding_idx = PAD_IDX)
151-
self.pe = self.pos_encoder(dim)
151+
self.pe = self.pos_encoding(dim)
152152
self.layers = nn.ModuleList([self.layer(dim) for _ in range(num_layers)])
153153

154154
def forward(self, x):
@@ -161,7 +161,7 @@ def forward(self, x):
161161

162162
return h
163163

164-
def pos_encoder(self, dim, maxlen = 1000): # positional encoding
164+
def pos_encoding(self, dim, maxlen = 1000): # positional encoding
165165

166166
pe = Tensor(maxlen, dim)
167167
pos = torch.arange(0, maxlen, 1.).unsqueeze(1)

0 commit comments

Comments
 (0)