From 231c96f9f151371e15d6ea8ac706366ccd43ff3c Mon Sep 17 00:00:00 2001
From: Maciej Wielgosz <maciej.wielgosz@nibio.no>
Date: Fri, 24 Mar 2023 14:19:57 +0100
Subject: [PATCH] parallel implementation of dgcnn in pl

---
 dgcnn/dgcnn_train_pl.py | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/dgcnn/dgcnn_train_pl.py b/dgcnn/dgcnn_train_pl.py
index e6ba0c2..b891128 100644
--- a/dgcnn/dgcnn_train_pl.py
+++ b/dgcnn/dgcnn_train_pl.py
@@ -6,6 +6,8 @@ from shapenet_data_dgcnn import ShapenetDataDgcnn
 import pytorch_lightning as pl
 from pytorch_lightning.loggers import WandbLogger
 from model import DGCNN
+from pytorch_lightning.strategies import DDPStrategy
+
 
 
 
@@ -54,7 +56,7 @@ dataloader = torch.utils.data.DataLoader(
 
 wandb_logger = WandbLogger(project="dgcnn", name="dgcnn", entity="maciej-wielgosz-nibio")
 
-trainer = pl.Trainer(accelerator="auto", devices=[0], max_epochs=3, logger=wandb_logger, gpus=1)
+trainer = pl.Trainer(strategy=DDPStrategy(find_unused_parameters=True), accelerator="auto", devices=[0], max_epochs=3, logger=wandb_logger, gpus=1)
 
 # Initialize a model
 model = DGCNNLightning(num_classes=16)
-- 
GitLab