note for DDP inference (#1194)

Co-authored-by: Ethan Harris <[email protected]>
Lightning-Universe · Mar 1, 2022 · 8abc10a · 8abc10a
1 parent 8144087
commit 8abc10a
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/docs/source/general/predictions.rst b/docs/source/general/predictions.rst
@@ -66,3 +66,7 @@ reference below).
     predictions = trainer.predict(model, datamodule=datamodule)
     print(predictions)
     # out: [[[0.5926494598388672, 0.40735048055648804]]]
+
+
+.. note::
+    PyTorch Lightning does not return predictions directly from `predict` when using a multi-GPU configuration (DDP). Instead you should use a :class:`pytorch_lightning.callbacks.BasePredictionWriter`.