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)