Skip to content

Commit 4b12bd1

Browse files
MapDatasetWrapper: support for external seq_order
1 parent 0b73a90 commit 4b12bd1

File tree

1 file changed

+4
-1
lines changed

1 file changed

+4
-1
lines changed

returnn/datasets/map.py

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -102,8 +102,11 @@ def init_seq_order(self, epoch=None, seq_list=None, seq_order=None):
102102
"""
103103
super(MapDatasetWrapper, self).init_seq_order(epoch=epoch, seq_list=seq_list, seq_order=seq_order)
104104

105-
if seq_list or seq_order:
105+
if seq_list:
106106
raise NotImplementedError
107+
if seq_order:
108+
self._seq_order = seq_order
109+
return True
107110

108111
try:
109112
self._seq_order = self._dataset.get_seq_order(epoch=epoch)

0 commit comments

Comments
 (0)