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.
2 parents c4dbc7b + 29fe62a commit 281b567Copy full SHA for 281b567
fastNLP/core/batch.py
@@ -82,6 +82,6 @@ def __next__(self):
82
batch_origin_length[name + "_origin_len"] = torch.LongTensor(origin_lengths[name])
83
batch_x.update(batch_origin_length)
84
85
- self.curidx += endidx
+ self.curidx = endidx
86
return batch_x, batch_y
87
0 commit comments