update

Lightning-Universe · Dec 8, 2020 · 2e3199b · 2e3199b
1 parent 4074654
commit 2e3199b
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/pl_bolts/callbacks/variational.py b/pl_bolts/callbacks/variational.py
@@ -33,15 +33,16 @@ def __init__(
         range_end: int = 5,
         steps: int = 11,
         num_samples: int = 2,
-        normalize=False,
+        normalize: bool = True,
     ):
         """
         Args:
             interpolate_epoch_interval: default 20
             range_start: default -5
             range_end: default 5
+            steps: number of step between start and end
             num_samples: default 2
-            normalize: default False
+            normalize: default True (change image to (0, 1) range)
         """
         super().__init__()
         self.interpolate_epoch_interval = interpolate_epoch_interval