returnn 1.20240720.235853__tar.gz → 1.20240721.1359__tar.gz

{returnn-1.20240720.235853 → returnn-1.20240721.1359}/PKG-INFO

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

returnn-1.20240721.1359/_setup_info_generated.py

@@ -0,0 +1,2 @@
+version = '1.20240721.001359'
+long_version = '1.20240721.001359+git.41c3a52'

{returnn-1.20240720.235853 → returnn-1.20240721.1359}/returnn.egg-info/PKG-INFO

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

{returnn-1.20240720.235853 → returnn-1.20240721.1359}/tests/test_torch_frontend.py

@@ -627,6 +627,7 @@ def _benchmark_pack_padded():
         dims=[vocab_dim.copy(match_priority=1), vocab_dim],
     )
     sizeof_float = 4
+    print("logits shape:", logits.raw_tensor.shape)
     print("logits size:", logits.raw_tensor.numel() * sizeof_float, "bytes")
     print("dev:", logits.device)
 
@@ -714,7 +715,10 @@ def _benchmark_pack_padded():
     ]:
         print("func:", f.__name__)
         t = Timer(stmt="func()", globals={"func": f})
-        print(t.blocked_autorange(min_run_time=0.5))
+        try:
+            print(t.blocked_autorange(min_run_time=0.5))
+        except torch.cuda.OutOfMemoryError:
+            print("CUDA OOM")
 
 
 def _benchmark_pack_padded_one_dim():
@@ -745,6 +749,7 @@ def _benchmark_pack_padded_one_dim():
         dims=[batch_dim_, enc_dim, vocab_dim],
     )
     sizeof_float = 4
+    print("logits shape:", logits.raw_tensor.shape)
     print("logits size:", logits.raw_tensor.numel() * sizeof_float, "bytes")
     print("dev:", logits.device)
 
@@ -844,7 +849,10 @@ def _benchmark_pack_padded_one_dim():
     ]:
         print("func:", f.__name__)
         t = Timer(stmt="func()", globals={"func": f})
-        print(t.blocked_autorange(min_run_time=0.5))
+        try:
+            print(t.blocked_autorange(min_run_time=0.5))
+        except torch.cuda.OutOfMemoryError:
+            print("CUDA OOM")
 
 
 def test_Data_copy_compatible_to_match_priority():

returnn-1.20240720.235853/_setup_info_generated.py

@@ -1,2 +0,0 @@
-version = '1.20240720.235853'
-long_version = '1.20240720.235853+git.4b5e39c'