complexhalf (complex32) isn't supported by torch.linalg.qr

2024-11-20 22:56:26 +01:00
parent 6358c95c42
commit 8d4d0468bd
1 changed files with 1 additions and 1 deletions
--- a/src/single-core-regen/regen.py
+++ b/src/single-core-regen/regen.py
@@ -15,7 +15,7 @@ global_settings = GlobalSettings(

 data_settings = DataSettings(
    config_path     = "data/*-128-16384-100000-0-0-17-0-PAM4-0.ini",
-    dtype           = ("complex64", "float64", "complex32", "float32"),
+    dtype           = ("complex128", "complex64", "float64", "float32"),
    symbols         = (1, 16),
    model_input_dim = (1, 32),
    shuffle         = True,