hummingbird
/
fastNLP
Not watched
2
0
Fork 0
Code
Releases 13 Wiki Activity
Issues 0 Pull Requests 0 Datasets Model Cloudbrain
Browse Source

Merge pull request #90 from fastnlp/fix-batch 

fix bug
tags/v0.1.0
Yige XU GitHub 6 years ago
parent
c4dbc7b902 29fe62a959
commit
281b5671fb
No known key found for this signature in database GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 1 deletions
Split View
Diff Options
Show Stats Download Patch File Download Diff File
  1. +1
    -1
      fastNLP/core/batch.py

+ 1
- 1
fastNLP/core/batch.py View File

@@ -82,6 +82,6 @@ class Batch(object):
batch_origin_length[name + "_origin_len"] = torch.LongTensor(origin_lengths[name])
batch_x.update(batch_origin_length)

self.curidx += endidx
self.curidx = endidx
return batch_x, batch_y


Write Preview
Loading…
Cancel
Save