minor bugfix

Author: jinserk

asr/models/las/network.py

@@ -374,7 +374,7 @@ def _train_forward(self, x, x_seq_lens, y, y_seq_lens):
         eos_t = y.new_full((self.spell.num_eos, ), self.eos)
         ys = [torch.cat((yb, eos_t)) for yb in torch.split(y, y_seq_lens.tolist())]
         ys = nn.utils.rnn.pad_sequence(ys, batch_first=True, padding_value=self.blk)
-        ys, ys_seq_lens = ys[bi], y_seq_lens[bi] + 1
+        ys, ys_seq_lens = ys[bi], y_seq_lens[bi] + self.spell.num_eos
 
         if self._is_teacher_force():
             # speller with teach force rate including noise

requirements.txt

@@ -1,8 +1,10 @@
 numpy
 scipy
-torch_nightly --find-links https://download.pytorch.org/whl/nightly/cpu/torch_nightly.html
+#torch_nightly --find-links https://download.pytorch.org/whl/nightly/cpu/torch_nightly.html
+torch -f https://download.pytorch.org/whl/cu100/torch-1.0.0-cp37-cp37m-linux_x86_64.whl
 git+https://github.com/pytorch/audio
-git+https://github.com/pytorch/vision
+#git+https://github.com/pytorch/vision
+torchvision
 git+https://github.com/pytorch/text
 git+https://github.com/pytorch/tnt
 git+https://github.com/NVIDIA/apex