resolve bug

Lightning-Universe · Dec 1, 2021 · 9c3d408 · 9c3d408
1 parent a2c5034
commit 9c3d408
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/flash/image/style_transfer/data.py b/flash/image/style_transfer/data.py
@@ -93,7 +93,7 @@ def default_transforms(self) -> Optional[Dict[str, Callable]]:
                 ),
             )
         if self.predicting:
-            return dict(per_sample_transform=nn.Sequential(T.Resize(self.image_size), T.ToTensor()))
+            return dict(per_sample_transform=T.Compose([T.Resize(self.image_size), T.ToTensor()]))
         # Style transfer doesn't support a validation or test phase, so we return nothing here
         return None