Skip to content

Commit fb8379a

Browse files
TQuysoumith
authored andcommitted
fix the warning when training (#673)
change ByteTensor to BoolTensor
1 parent aec0988 commit fb8379a

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

beginner_source/chatbot_tutorial.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -537,7 +537,7 @@ def outputVar(l, voc):
537537
max_target_len = max([len(indexes) for indexes in indexes_batch])
538538
padList = zeroPadding(indexes_batch)
539539
mask = binaryMatrix(padList)
540-
mask = torch.ByteTensor(mask)
540+
mask = torch.BoolTensor(mask)
541541
padVar = torch.LongTensor(padList)
542542
return padVar, mask, max_target_len
543543

0 commit comments

Comments
 (0)