Followup of #843 - adding docstring (#847)

Lightning-Universe · Aug 2, 2022 · 5481b6e · 5481b6e
1 parent f3b17aa
commit 5481b6e
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/pl_bolts/datamodules/vision_datamodule.py b/pl_bolts/datamodules/vision_datamodule.py
@@ -45,6 +45,9 @@ def __init__(
             pin_memory: If true, the data loader will copy Tensors into CUDA pinned memory before
                         returning them
             drop_last: If true drops the last incomplete batch
+            train_transforms: transformations you can apply to train dataset
+            val_transforms: transformations you can apply to validation dataset
+            test_transforms: transformations you can apply to test dataset
         """
 
         super().__init__(*args, **kwargs)