We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent be8558f commit 6ade2bfCopy full SHA for 6ade2bf
src/11_seq2seq/modules/trainer.py
@@ -59,7 +59,7 @@ def train(engine, mini_batch):
59
그 중에서 실제 문장 데이터만 GPU 메모리로 전송
60
'''
61
mini_batch.src = (mini_batch.src[0].to(device), mini_batch.src[1])
62
- mini_batch.tgt = (mini_batch.src[0].to(device), mini_batch.tgt[1])
+ mini_batch.tgt = (mini_batch.tgt[0].to(device), mini_batch.tgt[1])
63
64
65
맨 처음 Input으로 x가 들어감
0 commit comments