More adjustments

This commit is contained in:
henryruhs
2025-03-11 14:43:10 +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')