CPU and GPU trainings.

usnistgov · Apr 28, 2024 · 6b11030 · 6b11030
1 parent 114cc9e
commit 6b11030
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/alignn/train.py b/alignn/train.py
@@ -195,7 +195,7 @@ def train_dgl(
             standard_scalar_and_pca=config.standard_scalar_and_pca,
             keep_data_order=config.keep_data_order,
             output_dir=config.output_dir,
-            use_ddp=use_ddp,
+            # use_ddp=use_ddp,
         )
     else:
         train_loader = train_val_test_loaders[0]