returnn 1.20250404.233250__py3-none-any.whl → 1.20250405.2748__py3-none-any.whl

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.

Potentially problematic release.


This version of returnn might be problematic. Click here for more details.

returnn/PKG-INFO CHANGED
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.1
2
2
  Name: returnn
3
- Version: 1.20250404.233250
3
+ Version: 1.20250405.2748
4
4
  Summary: The RWTH extensible training framework for universal recurrent neural networks
5
5
  Home-page: https://github.com/rwth-i6/returnn/
6
6
  Author: Albert Zeyer
@@ -1,2 +1,2 @@
1
- version = '1.20250404.233250'
2
- long_version = '1.20250404.233250+git.1aa10a4'
1
+ version = '1.20250405.002748'
2
+ long_version = '1.20250405.002748+git.cf50800'
returnn/torch/engine.py CHANGED
@@ -942,8 +942,13 @@ class Engine(EngineBase):
942
942
  continue
943
943
  if opts["filename"] is None:
944
944
  print(f"Pre-load (initialize) weights for key '{preload_key}'", file=log.v3)
945
- pattern = opts["pattern"]
946
- match = re.compile(fnmatch.translate(pattern)).match
945
+ if opts.get("pattern", None) is not None:
946
+ pattern = opts["pattern"]
947
+ match = re.compile(fnmatch.translate(pattern)).match
948
+ elif opts.get("prefix", None) is not None:
949
+ match = re.compile(re.escape(opts["prefix"]) + ".*").fullmatch
950
+ else:
951
+ raise ValueError(f"preload key {preload_key} without file {opts}: no pattern or prefix given")
947
952
  remove = []
948
953
  for name in self._pt_model.state_dict().keys():
949
954
  if match(name) and name in missing_keys:
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.1
2
2
  Name: returnn
3
- Version: 1.20250404.233250
3
+ Version: 1.20250405.2748
4
4
  Summary: The RWTH extensible training framework for universal recurrent neural networks
5
5
  Home-page: https://github.com/rwth-i6/returnn/
6
6
  Author: Albert Zeyer
@@ -1,9 +1,9 @@
1
- returnn/PKG-INFO,sha256=qejKOAnhXlw-GA7gM-OP-HG1SgkdQr7PWfYnrciKkDk,5215
1
+ returnn/PKG-INFO,sha256=K0md_kksJ1J_WrXSBPCiZ7NWu8gqvXkceUf6El-JAOQ,5213
2
2
  returnn/__init__.py,sha256=biBtRsM0WZ406vShaeH-9WFoqJ8XwTbn6g0EeFJ7l8E,1012
3
3
  returnn/__main__.py,sha256=qBFbuB1yN3adgVM5pXt2-Yq9vorjRNchNPL8kDKx44M,31752
4
4
  returnn/__old_mod_loader__.py,sha256=nvsNY-xELdS_IPNkv66Q9Rmvg4dbGW0-EBRDcCmctos,7654
5
5
  returnn/__setup__.py,sha256=22kQn2fh11iPM0hLb2Fy5sLmoU1JGvmDxXRYuRgQkwU,4659
6
- returnn/_setup_info_generated.py,sha256=bU1rN_PnvhlqxsSdtkgwp0Hx6GDEDnewK3fiT7NeTkk,77
6
+ returnn/_setup_info_generated.py,sha256=RGt81iFRXKiRnKcmYzasmg2bPzNGVN2aueyn97NFOQw,77
7
7
  returnn/config.py,sha256=3tmKhB6FnQZaNdtcYsiB61JnEY--iZ2qmJ4yq0b6tE0,29140
8
8
  returnn/forward_iface.py,sha256=A_OJiaXsX4MlXQRzST86ylyxSUZbC402PQL1REcqHjM,911
9
9
  returnn/learning_rate_control.py,sha256=ZvWryAn_tv9DhV8sh1LV3eE34Yltl3On3mYZAG4hR9s,34684
@@ -207,7 +207,7 @@ returnn/tf/util/open_fst.py,sha256=sZRDw4TbxvhGqpGdUJWy1ebvlZm4_RPhygpRw9uLAOQ,1
207
207
  returnn/torch/README.md,sha256=jzJ2FpOHW02vxN69yKaV97C9LI-hmvjBglKfdZXIDdc,85
208
208
  returnn/torch/__init__.py,sha256=MHEUyNHB20Vy89uKAqZoj6FxJKF1Gq3HW-i6ra1pNcI,24
209
209
  returnn/torch/distributed.py,sha256=skFyutdVztxgTEk3HHJ8S83qRWbNpkNT8Tj16Ic0_hE,6981
210
- returnn/torch/engine.py,sha256=2FLLb2m4sWFwYOQGREDSxQCheCKd_osnFJCdLa_4TzE,76400
210
+ returnn/torch/engine.py,sha256=HoQk9tSLRSPJpkTFv3HNkgsFGYHUWIa4dHpRjvAtGgQ,76765
211
211
  returnn/torch/updater.py,sha256=GqtBvZpElPVMm0lq84JPl4NVLFFETZAzAbR0rTomSao,28249
212
212
  returnn/torch/data/__init__.py,sha256=6cLNEi8KoGI12PF6akN7mI_mtjlx-0hcQAfMYoExwik,132
213
213
  returnn/torch/data/extern_data.py,sha256=OSoy3x1KiyiJCr7DfF5uPFAu09We2N2WbA0yo-pYXxM,7601
@@ -253,8 +253,8 @@ returnn/util/sig_proc.py,sha256=Tjz0VOAVyqu2qDCF5HZ1JjALjcFsHcNkcd96WgZeKfE,7265
253
253
  returnn/util/task_system.py,sha256=y4sMVXQ25Qd2z0rx03uOlXlkE-jbCYC1Sjfn-XlraVU,26003
254
254
  returnn/util/train_proc_manager.py,sha256=Pjht28k6uz6BNQ47uW6Gf880iyq5q4wx7P_K2tmoAM8,3266
255
255
  returnn/util/watch_memory.py,sha256=BR5P2kvBN6UI81cE0_1WAA6Hd1SByLbBaiDxvLhPOew,4213
256
- returnn-1.20250404.233250.dist-info/LICENSE,sha256=ywBD_U2aD4vpuoIgNAsjIGBYydl0tVKll3De0Z8s77c,11041
257
- returnn-1.20250404.233250.dist-info/METADATA,sha256=qejKOAnhXlw-GA7gM-OP-HG1SgkdQr7PWfYnrciKkDk,5215
258
- returnn-1.20250404.233250.dist-info/WHEEL,sha256=iAkIy5fosb7FzIOwONchHf19Qu7_1wCWyFNR5gu9nU0,91
259
- returnn-1.20250404.233250.dist-info/top_level.txt,sha256=Lsn4WZc5Pbfk0-xDQOgnFCxOoqxL4CyeM3N1TFbJncw,8
260
- returnn-1.20250404.233250.dist-info/RECORD,,
256
+ returnn-1.20250405.2748.dist-info/LICENSE,sha256=ywBD_U2aD4vpuoIgNAsjIGBYydl0tVKll3De0Z8s77c,11041
257
+ returnn-1.20250405.2748.dist-info/METADATA,sha256=K0md_kksJ1J_WrXSBPCiZ7NWu8gqvXkceUf6El-JAOQ,5213
258
+ returnn-1.20250405.2748.dist-info/WHEEL,sha256=iAkIy5fosb7FzIOwONchHf19Qu7_1wCWyFNR5gu9nU0,91
259
+ returnn-1.20250405.2748.dist-info/top_level.txt,sha256=Lsn4WZc5Pbfk0-xDQOgnFCxOoqxL4CyeM3N1TFbJncw,8
260
+ returnn-1.20250405.2748.dist-info/RECORD,,