parallel implementation of dgcnn in pl

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)