scaling image data

tsugumi-sys · Jan 15, 2024 · 6504fc4 · 6504fc4
1 parent 9883b9e
commit 6504fc4
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/data_loaders/moving_mnist.py b/data_loaders/moving_mnist.py
@@ -56,7 +56,9 @@ def __init__(
             )
         self.split_ratios = split_ratios
 
-        moving_mnist = MovingMNIST(root="./data", download=True)
+        moving_mnist = MovingMNIST(
+            root="./data", download=True, transform=lambda x: x / 255.0
+        )
         train_dataset, valid_dataset, test_dataset = random_split(
             moving_mnist,
             [*self.split_ratios],