Update trainer.py

metatensor · Jul 18, 2024 · df9bd2d · df9bd2d
1 parent e858dcd
commit df9bd2d
Showing 1 changed file with 0 additions and 3 deletions.
diff --git a/src/metatrain/experimental/pet/trainer.py b/src/metatrain/experimental/pet/trainer.py
@@ -80,9 +80,6 @@ def train(
 
         # set model hypers
         self.hypers["ARCHITECTURAL_HYPERS"] = model.hypers
-        dtype = train_datasets[0][0]["system"].positions.dtype
-        if dtype != torch.float32:
-            raise ValueError("PET only supports float32 as dtype")
         self.hypers["ARCHITECTURAL_HYPERS"]["DTYPE"] = "float32"
 
         # set MLIP_SETTINGS