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 +1 -1
- returnn/_setup_info_generated.py +2 -2
- returnn/torch/engine.py +7 -2
- {returnn-1.20250404.233250.dist-info → returnn-1.20250405.2748.dist-info}/METADATA +1 -1
- {returnn-1.20250404.233250.dist-info → returnn-1.20250405.2748.dist-info}/RECORD +8 -8
- {returnn-1.20250404.233250.dist-info → returnn-1.20250405.2748.dist-info}/LICENSE +0 -0
- {returnn-1.20250404.233250.dist-info → returnn-1.20250405.2748.dist-info}/WHEEL +0 -0
- {returnn-1.20250404.233250.dist-info → returnn-1.20250405.2748.dist-info}/top_level.txt +0 -0
returnn/PKG-INFO
CHANGED
returnn/_setup_info_generated.py
CHANGED
|
@@ -1,2 +1,2 @@
|
|
|
1
|
-
version = '1.
|
|
2
|
-
long_version = '1.
|
|
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
|
-
|
|
946
|
-
|
|
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,9 +1,9 @@
|
|
|
1
|
-
returnn/PKG-INFO,sha256=
|
|
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=
|
|
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=
|
|
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.
|
|
257
|
-
returnn-1.
|
|
258
|
-
returnn-1.
|
|
259
|
-
returnn-1.
|
|
260
|
-
returnn-1.
|
|
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,,
|
|
File without changes
|
|
File without changes
|
|
File without changes
|