fix(torch): handle case where sequence data is < wanted timestep

jolibrain · Sep 14, 2020 · b6d394a · b6d394a
1 parent 81f47c9
commit b6d394a
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/backends/torch/torchinputconns.cc b/src/backends/torch/torchinputconns.cc
@@ -725,7 +725,7 @@ namespace dd
             at::Tensor lst = torch::stack(label_sequence);
             dataset.add_batch({ dst }, { lst });
           }
-        if (dv.rem != 0)
+        if (dv.rem != 0 && seq.size() >= _timesteps)
           {
             std::vector<at::Tensor> data_sequence;
             std::vector<at::Tensor> label_sequence;