added .csv image loading utils

Lightning-Universe · Feb 14, 2021 · f435f8f · f435f8f
1 parent b8b52e3
commit f435f8f
Showing 1 changed file with 5 additions and 3 deletions.
diff --git a/flash/vision/classification/data.py b/flash/vision/classification/data.py
@@ -329,9 +329,11 @@ def from_filepaths(
             full_length = len(train_ds)
             train_split = int((1.0 - valid_split) * full_length)
             valid_split = full_length - train_split
-            train_ds, valid_ds = torch.utils.data.random_split(train_ds,
-                                                               [train_split, valid_split],
-                                                               generator=torch.Generator().manual_seed(seed))
+            train_ds, valid_ds = torch.utils.data.random_split(
+                train_ds,
+                [train_split, valid_split],
+                generator=torch.Generator().manual_seed(seed)
+            )
         else:
             valid_ds = (
                 FilepathDataset(