Use strategy instead of plugins for DDP

requirements-dev/docs.txt

@@ -1,4 +1,4 @@
 mike
 mkdocs>=1.2.2
 mkdocs-material>=7.3.4
-mkdocstrings>=0.15.2
+mkdocstrings[python]>=0.19.0

setup.cfg

@@ -31,6 +31,7 @@ install_requires =
     openpyxl
     pandas>1.2.0
     pandas_path
+    protobuf==3.20.1
     pydantic
     python-dotenv
     pytorch-lightning>=1.6.0

zamba/models/config.py

@@ -451,7 +451,7 @@ def validate_filepaths_and_labels(cls, values):
 
             elif values["split_proportions"] is not None:
                 logger.warning(
-                    "Labels contains split column yet split_proprtions are also provided. Split column in labels takes precendece."
+                    "Labels contains split column yet split_proportions are also provided. Split column in labels takes precedence."
                 )
                 # set to None for clarity in final configuration.yaml
                 values["split_proportions"] = None

zamba/models/model_manager.py

@@ -268,8 +268,7 @@ def train_model(
         logger=tensorboard_logger,
         callbacks=callbacks,
         fast_dev_run=train_config.dry_run,
-        accelerator="ddp" if data_module.multiprocessing_context is not None else None,
-        plugins=DDPPlugin(find_unused_parameters=False)
+        strategy=DDPPlugin(find_unused_parameters=False)
         if data_module.multiprocessing_context is not None
         else None,
     )