change trainer strategy to "auto" (#234)

torchmd · Nov 14, 2023 · c3b46f6 · c3b46f6
1 parent a67ac80
commit c3b46f6
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/torchmdnet/scripts/train.py b/torchmdnet/scripts/train.py
@@ -173,7 +173,7 @@ def main():
         )
 
     trainer = pl.Trainer(
-        strategy=DDPStrategy(find_unused_parameters=False),
+        strategy="auto",
         max_epochs=args.num_epochs,
         accelerator="auto",
         devices=args.ngpus,