Commit 4d03d505 by libei

remove some logs

parent 68deabe3
...@@ -127,13 +127,13 @@ class TransformerEncoder(FairseqEncoder): ...@@ -127,13 +127,13 @@ class TransformerEncoder(FairseqEncoder):
emb_token = self.embed_tokens(src_tokens) emb_token = self.embed_tokens(src_tokens)
# print('emb_token:', emb_token[0]) # print('emb_token:', emb_token[0])
x = self.embed_scale * emb_token x = self.embed_scale * emb_token
print('scaled_word_emb:', x[0]) # print('scaled_word_emb:', x[0])
pos_emb = self.embed_positions(src_tokens) pos_emb = self.embed_positions(src_tokens)
# print('pos:',pos_emb.size()) # print('pos:',pos_emb.size())
print('pos:', pos_emb) # print('pos:', pos_emb)
x += pos_emb x += pos_emb
# print('final emb:', x.size()) # print('final emb:', x.size())
print('final emb:', x[0]) # print('final emb:', x[0])
#raise #raise
x = F.dropout(x, p=self.dropout, training=self.training) x = F.dropout(x, p=self.dropout, training=self.training)
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论