returnn 1.20230620.175850__tar.gz → 1.20230626.20229__tar.gz

{returnn-1.20230620.175850/returnn.egg-info → returnn-1.20230626.20229}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: returnn
-Version: 1.20230620.175850
+Version: 1.20230626.20229
 Summary: The RWTH extensible training framework for universal recurrent neural networks
 Home-page: https://github.com/rwth-i6/returnn/
 Author: Albert Zeyer

returnn-1.20230626.20229/_setup_info_generated.py

@@ -0,0 +1,2 @@
+version = '1.20230626.020229'
+long_version = '1.20230626.020229+git.283f125'

{returnn-1.20230620.175850 → returnn-1.20230626.20229}/returnn/datasets/meta.py

@@ -427,6 +427,13 @@ class MetaDataset(CachedDataset2):
             dataset.init_seq_order(epoch=epoch, seq_list=self.seq_list_ordered[dataset_key])
         return True
 
+    def get_current_seq_order(self):
+        """
+        :return: current seq order for the current epoch, after self.init_seq_order was called.
+        :rtype: list[int]
+        """
+        return [self.tag_idx[tag] for tag in self.seq_list_ordered[self.default_dataset_key]]
+
     def get_all_tags(self):
         """
         :return: list of all seq tags, of the whole dataset, without partition epoch

{returnn-1.20230620.175850 → returnn-1.20230626.20229/returnn.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: returnn
-Version: 1.20230620.175850
+Version: 1.20230626.20229
 Summary: The RWTH extensible training framework for universal recurrent neural networks
 Home-page: https://github.com/rwth-i6/returnn/
 Author: Albert Zeyer

{returnn-1.20230620.175850 → returnn-1.20230626.20229}/tools/torch_export_to_onnx.py

@@ -11,6 +11,7 @@ from typing import Callable, Optional, Dict
 import argparse
 import os
 
+import _setup_returnn_env  # noqa
 from returnn.config import Config
 from returnn.log import log
 from returnn.tensor import TensorDict
@@ -117,12 +118,17 @@ class ForwardModuleRF(_RFModuleAsPTModule):
 def _check_matching_outputs():
     rf.get_run_ctx().check_outputs_complete()
     model_outputs_raw_keys = set(_get_model_outputs_raw_keys())
-    outputs_raw_keys = set(rf.get_run_ctx().outputs.as_raw_tensor_dict().keys())
+    outputs_raw = rf.get_run_ctx().outputs.as_raw_tensor_dict()
+    outputs_raw_keys = set(outputs_raw.keys())
     assert model_outputs_raw_keys == outputs_raw_keys, (
         f"Model outputs raw keys and output raw keys from forward_step don't match.\n"
         f"Model outputs raw keys: {sorted(model_outputs_raw_keys)}\n"
         f"Output raw keys: {sorted(outputs_raw_keys)}"
     )
+    assert all(v is not None for k, v in outputs_raw.items()), (
+        f"Output raw keys from forward_step contain None values.\n"
+        f"Output raw keys with None: {list(k for k, v in outputs_raw.items() if v is None)}"
+    )
 
 
 def _get_model_outputs_raw_keys():

returnn-1.20230620.175850/_setup_info_generated.py

@@ -1,2 +0,0 @@
-version = '1.20230620.175850'
-long_version = '1.20230620.175850+git.776bb1b'