stricter assertion

rwth-i6 · Jan 9, 2025 · 1bf5c8d · 1bf5c8d
1 parent 78ac8e5
commit 1bf5c8d
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/returnn/torch/data/returnn_dataset_wrapper.py b/returnn/torch/data/returnn_dataset_wrapper.py
@@ -116,7 +116,7 @@ def __iter__(self) -> Iterable[Dict[str, numpy.ndarray]]:
                 data["epoch"] = epoch
                 try:
                     epoch_cont = self._dataset.get_epoch_continuous(seq_index)
-                    assert epoch_cont < 1.0
+                    assert 0.0 <= epoch_cont < 1.0
                 except OptionalNotImplementedError:
                     epoch_cont = -1
                 data["epoch_continuous"] = numpy.array(epoch_cont)