Merge pull request #327 from RaulPPelaez/disable_tf

Disable TF32 usage by default
torchmd · May 21, 2024 · a6a7efb · a6a7efb
2 parents c9d55fc + 0f49ca1
commit a6a7efb
Showing 1 changed file with 0 additions and 2 deletions.
diff --git a/torchmdnet/models/tensornet.py b/torchmdnet/models/tensornet.py
@@ -13,8 +13,6 @@
 )
 
 __all__ = ["TensorNet"]
-torch.set_float32_matmul_precision("high")
-torch.backends.cuda.matmul.allow_tf32 = True
 
 
 def vector_to_skewtensor(vector):