Skip to content

Commit 6ade2bf

Browse files
committed
Update trainer.py
1 parent be8558f commit 6ade2bf

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

src/11_seq2seq/modules/trainer.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -59,7 +59,7 @@ def train(engine, mini_batch):
5959
그 중에서 실제 문장 데이터만 GPU 메모리로 전송
6060
'''
6161
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])
62+
mini_batch.tgt = (mini_batch.tgt[0].to(device), mini_batch.tgt[1])
6363

6464
'''
6565
맨 처음 Input으로 x가 들어감

0 commit comments

Comments
 (0)