From d74e1f5fcbcbc9f2bb610f4e45a7d4675aefeea3 Mon Sep 17 00:00:00 2001 From: Hanusz Leszek Date: Fri, 19 Aug 2022 14:51:00 +0200 Subject: [PATCH] Revert "Test with unclean file" This reverts commit ec46e153d9336b2010f2b6ce23fbca9ccca5577e. --- clean/04_mnist_basics.ipynb | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/clean/04_mnist_basics.ipynb b/clean/04_mnist_basics.ipynb index 6cc0bbf..a75eb39 100644 --- a/clean/04_mnist_basics.ipynb +++ b/clean/04_mnist_basics.ipynb @@ -174,7 +174,7 @@ "metadata": {}, "outputs": [], "source": [ - "stacked_sevens = torch.stack(seven_tensors)/255\n", + "stacked_sevens = torch.stack(seven_tensors).float()/255\n", "stacked_threes = torch.stack(three_tensors).float()/255\n", "stacked_threes.shape" ]