returnn 1.20230404.171419__tar.gz → 1.20230404.175609__tar.gz

{returnn-1.20230404.171419/returnn.egg-info → returnn-1.20230404.175609}/PKG-INFO

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

returnn-1.20230404.175609/_setup_info_generated.py

@@ -0,0 +1,2 @@
+version = '1.20230404.175609'
+long_version = '1.20230404.175609+git.e6dbb0e'

{returnn-1.20230404.171419 → returnn-1.20230404.175609}/returnn/tf/engine.py

@@ -860,11 +860,11 @@ class Engine(EngineBase):
         """
         super(Engine, self).__init__(config=config)
         if not log.initialized:
-            log.init_by_config(config)
+            log.init_by_config(self.config)
         if not BehaviorVersion.is_set():
-            BehaviorVersion.set(config.int("behavior_version", None))
+            BehaviorVersion.set(self.config.int("behavior_version", None))
         if BackendEngine.selected_engine is None:
-            BackendEngine.select_engine(engine=BackendEngine.TensorFlowNetDict)
+            BackendEngine.select_engine(default_fallback_engine=BackendEngine.TensorFlow, config=self.config)
         assert BackendEngine.is_tensorflow_selected()
         self.orig_config = {}  # see _maybe_update_config
         self.custom_get_net_dict = None  # type: typing.Optional[typing.Callable]
@@ -882,8 +882,8 @@ class Engine(EngineBase):
         self._num_trained_epochs = 0  # type: int  # just a counter
         self._num_net_reinit = 0  # type: int
         self.use_dynamic_train_flag = False
-        self.use_search_flag = config.value("task", None) == "search"
-        self.use_eval_flag = config.value("task", None) != "forward"
+        self.use_search_flag = self.config.value("task", None) == "search"
+        self.use_eval_flag = self.config.value("task", None) != "forward"
         self._const_cache = {}  # type: typing.Dict[str,tf.Tensor]
         self.preload_from_files = None  # type: typing.Optional[typing.Dict[str,typing.Dict[str]]]
         self.max_seqs = None  # type: typing.Optional[int]

{returnn-1.20230404.171419 → returnn-1.20230404.175609}/returnn/util/basic.py

@@ -127,9 +127,10 @@ class BackendEngine:
         """
 
     @classmethod
-    def select_engine(cls, engine=None, config=None):
+    def select_engine(cls, *, engine=None, default_fallback_engine=None, config=None):
         """
         :param int engine: see the global class attribs for possible values
+        :param int|None default_fallback_engine: if engine is None and not defined in config, use this
         :param returnn.config.Config config:
         """
         assert cls.selected_engine is None, "already set"
@@ -149,7 +150,10 @@ class BackendEngine:
                     "torch": cls.Torch,
                 }[backend]
             if engine is None:
-                engine = cls._get_default_engine()
+                if default_fallback_engine is not None:
+                    engine = default_fallback_engine
+                else:
+                    engine = cls._get_default_engine()
 
         # noinspection PyProtectedMember
         from returnn.frontend import _backend

{returnn-1.20230404.171419 → returnn-1.20230404.175609/returnn.egg-info}/PKG-INFO

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

returnn-1.20230404.171419/_setup_info_generated.py

@@ -1,2 +0,0 @@
-version = '1.20230404.171419'
-long_version = '1.20230404.171419+git.216ea5e'