returnn 1.20250821.93927__py3-none-any.whl → 1.20250826.155029__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.20250821.93927
3
+ Version: 1.20250826.155029
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.20250821.093927'
2
- long_version = '1.20250821.093927+git.ec56958'
1
+ version = '1.20250826.155029'
2
+ long_version = '1.20250826.155029+git.cca4212'
returnn/torch/updater.py CHANGED
@@ -440,8 +440,6 @@ class Updater:
440
440
  If it's a str, it must be the optimizer name.
441
441
  :return: tuple (optimizer, optional optimizer_param_groups_extra_opts).
442
442
  """
443
- lr = self.learning_rate
444
-
445
443
  # If the parameter is already a valid optimizer, return it without further processing
446
444
  if isinstance(optimizer_opts, torch.optim.Optimizer):
447
445
  return optimizer_opts, None
@@ -467,8 +465,9 @@ class Updater:
467
465
  opt_kwargs["eps"] = opt_kwargs.pop("epsilon")
468
466
  if "learning_rate" in opt_kwargs or "lr" in opt_kwargs:
469
467
  raise ValueError("'learning_rate' should be set outside of the 'optimizer' dict.")
470
- lr = lr * opt_kwargs.pop("learning_rate_multiplier", 1.0)
471
- opt_kwargs["lr"] = lr
468
+ # lr will anyway be updated in set_current_train_step / _update_effective_learning_rate,
469
+ # so this value doesn't really matter here
470
+ opt_kwargs["lr"] = self.learning_rate
472
471
 
473
472
  param_groups = self._get_optimizer_param_groups(optim_class, opt_kwargs)
474
473
  param_groups = list(param_groups)
@@ -390,6 +390,10 @@ class FileCache:
390
390
  # - https://github.com/rwth-i6/returnn/pull/1709
391
391
  os.utime(dst_filename, None)
392
392
  os.utime(info_file_name, None)
393
+ # Ensure we proactively make space for other users
394
+ # even in case we have all files ready on disk.
395
+ # See for discussion: https://github.com/rwth-i6/returnn/pull/1752.
396
+ self.cleanup(need_at_least_free_space_size=0)
393
397
  return
394
398
 
395
399
  print(f"FileCache: Copy file {src_filename} to cache")
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.1
2
2
  Name: returnn
3
- Version: 1.20250821.93927
3
+ Version: 1.20250826.155029
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=oANuix-AgPHTYt9t0MfUfM0PyjXdMuEBJrN2b99tMUI,5214
1
+ returnn/PKG-INFO,sha256=ruOzYwqndb2UaDGaPErhnL7kyTDRDm0J-R2DCrSeNyI,5215
2
2
  returnn/__init__.py,sha256=biBtRsM0WZ406vShaeH-9WFoqJ8XwTbn6g0EeFJ7l8E,1012
3
3
  returnn/__main__.py,sha256=lHyZcu_0yc9f7Vf_Kfdy9PmeU0T76XVXnpalHi5WKro,31740
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=o3ap30O-BqlF3l-gxibSvi0MgnNA3LJEG6o9FVoM0no,77
6
+ returnn/_setup_info_generated.py,sha256=ZrVDjVsQQnP6xCFRVMbLAnFYhuL76CBWniasxrCMZSw,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
@@ -208,7 +208,7 @@ 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=_lyJR71HIoCHpMi5GztGM7YwrX54Am8zSkjnDkE1Lbk,7524
210
210
  returnn/torch/engine.py,sha256=JSsQZZiVs9TxRyFEJuR3iH-YZb9sRw7TzoIAIqmplZY,78275
211
- returnn/torch/updater.py,sha256=7lMoA01Yzp18MY5jjIFncsajTjOD713pK38nU6r-jiE,31999
211
+ returnn/torch/updater.py,sha256=nNd1mBPQyvIB096BEFi0KKmRI-U3jnRETzb743p2B9c,32064
212
212
  returnn/torch/data/__init__.py,sha256=6cLNEi8KoGI12PF6akN7mI_mtjlx-0hcQAfMYoExwik,132
213
213
  returnn/torch/data/extern_data.py,sha256=5al706ZaYtHWLp5VH2vS-rW69YXP3NHyOFRKY0WY714,7810
214
214
  returnn/torch/data/pipeline.py,sha256=HgIL0jQsPcgvh_SPC4wQ6BzclmrnpFja-UiboF_GPN4,29459
@@ -238,7 +238,7 @@ returnn/util/better_exchook.py,sha256=39yvRecluDgYhViwSkaQ8crJ_cBWI63KeEGuK4RKe5
238
238
  returnn/util/bpe.py,sha256=LWFhICZsEOnMwNws0lybPNzKRX6rSr8yKCvP65vjl9Y,19656
239
239
  returnn/util/debug.py,sha256=wuRzdg9zB84WWCGyTjmRR_zYypu8gXxlc0nZ6si9OC8,28224
240
240
  returnn/util/debug_helpers.py,sha256=0EINLK4uLtoSt5_kHs1M2NIFpMd0S7i4c4rx90U4fJk,2914
241
- returnn/util/file_cache.py,sha256=15p0daE4KmzO5OvkhDr9cA_zJNbJrX98mnL-2fECtmM,27575
241
+ returnn/util/file_cache.py,sha256=ERGz6TEWqetGk4odj1x6cMfecfQ5G5G4e5psSrbx03Y,27852
242
242
  returnn/util/fsa.py,sha256=k2lJ8tyf_g44Xk1EPVLwDwpP4spoMTqIigDVOWocQHY,59177
243
243
  returnn/util/literal_py_to_pickle.py,sha256=3dnjWPeeiDT2xp4bRDgIf9yddx7b1AG7mOKEn_jiSl8,2173
244
244
  returnn/util/lru_cache.py,sha256=7Q5H3a8b07E8e1iB7PA9jCpRnxMJZOFS2KO07cy0gqk,11446
@@ -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.20250821.93927.dist-info/LICENSE,sha256=ywBD_U2aD4vpuoIgNAsjIGBYydl0tVKll3De0Z8s77c,11041
257
- returnn-1.20250821.93927.dist-info/METADATA,sha256=oANuix-AgPHTYt9t0MfUfM0PyjXdMuEBJrN2b99tMUI,5214
258
- returnn-1.20250821.93927.dist-info/WHEEL,sha256=iAkIy5fosb7FzIOwONchHf19Qu7_1wCWyFNR5gu9nU0,91
259
- returnn-1.20250821.93927.dist-info/top_level.txt,sha256=Lsn4WZc5Pbfk0-xDQOgnFCxOoqxL4CyeM3N1TFbJncw,8
260
- returnn-1.20250821.93927.dist-info/RECORD,,
256
+ returnn-1.20250826.155029.dist-info/LICENSE,sha256=ywBD_U2aD4vpuoIgNAsjIGBYydl0tVKll3De0Z8s77c,11041
257
+ returnn-1.20250826.155029.dist-info/METADATA,sha256=ruOzYwqndb2UaDGaPErhnL7kyTDRDm0J-R2DCrSeNyI,5215
258
+ returnn-1.20250826.155029.dist-info/WHEEL,sha256=iAkIy5fosb7FzIOwONchHf19Qu7_1wCWyFNR5gu9nU0,91
259
+ returnn-1.20250826.155029.dist-info/top_level.txt,sha256=Lsn4WZc5Pbfk0-xDQOgnFCxOoqxL4CyeM3N1TFbJncw,8
260
+ returnn-1.20250826.155029.dist-info/RECORD,,