Skip to content

Commit b3889bf

Browse files
committed
remove print
1 parent cf9f407 commit b3889bf

File tree

2 files changed

+2
-1
lines changed

2 files changed

+2
-1
lines changed

5-1.Transformer/Transformer(Greedy_decoder)-Torch.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -52,7 +52,7 @@ def get_posi_angle_vec(position):
5252
return torch.FloatTensor(sinusoid_table)
5353

5454
def get_attn_pad_mask(seq_q, seq_k):
55-
print(seq_q)
55+
# print(seq_q)
5656
batch_size, len_q = seq_q.size()
5757
batch_size, len_k = seq_k.size()
5858
# eq(zero) is PAD token

5-1.Transformer/Transformer(Greedy_decoder)_Torch.ipynb

+1
Original file line numberDiff line numberDiff line change
@@ -147,6 +147,7 @@
147147
" return torch.FloatTensor(sinusoid_table)\n",
148148
"\n",
149149
"def get_attn_pad_mask(seq_q, seq_k):\n",
150+
" # print(seq_q)\n",
150151
" batch_size, len_q = seq_q.size()\n",
151152
" batch_size, len_k = seq_k.size()\n",
152153
" # eq(zero) is PAD token\n",

0 commit comments

Comments
 (0)