Skip to content

GitLab

"git@developer.sourcefind.cn:renzhc/diffusers_dcu.git" did not exist on "256e0106749363fce06c28000698edeaf56a874d"
Commit 0ac24914 authored by serendipityCoding's avatar serendipityCoding
Browse files

fix typo in transformer example

parent 0ec96584
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
examples/transformer-xl/mem_transformer.py
View file @ 0ac24914
...@@ -513,7 +513,7 @@ class AdaptiveEmbedding(nn.Module): ...@@ -513,7 +513,7 @@ class AdaptiveEmbedding(nn.Module):
l_idx, r_idx = self.cutoff_ends[i], self.cutoff_ends[i+1] l_idx, r_idx = self.cutoff_ends[i], self.cutoff_ends[i+1]
d_emb_i = d_embed // (div_val ** i) d_emb_i = d_embed // (div_val ** i)
self.emb_layers.append(nn.Embedding(r_idx-l_idx, d_emb_i)) self.emb_layers.append(nn.Embedding(r_idx-l_idx, d_emb_i))
self.emb_projs.append(Projectio(d_proj, d_emb_i)) self.emb_projs.append(Projection(d_proj, d_emb_i))
def forward(self, inp): def forward(self, inp):
if self.div_val == 1: if self.div_val == 1:
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment