More adjustments

This commit is contained in:
henryruhs
2025-03-06 13:10:20 +01:00
parent 847579f925
commit 5bacb048dd
-1
View File
@@ -139,7 +139,6 @@ def train() -> None:
}
}
if torch.cuda.is_available():
torch.set_float32_matmul_precision('high')