feat(torch): in case of timeseries, warn if file do not contain enoug…

…h timesteps
jolibrain · Oct 5, 2020 · 1a5f905 · 1a5f905
1 parent d37e182
commit 1a5f905
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/src/backends/torch/torchinputconns.cc b/src/backends/torch/torchinputconns.cc
@@ -993,6 +993,13 @@ namespace dd
       {
         vecindex++;
         long int tstart = 0;
+        if (static_cast<long int>(seq.size()) < _timesteps)
+          {
+            _tilogger->warn(
+                "file " + _fnames[vecindex]
+                + " does not contains enough timesteps, discarding");
+            continue;
+          }
         for (; tstart + _timesteps < static_cast<long int>(seq.size());
              tstart += _offset)
           // construct timeseries here	, using timesteps and offset from data