Change image shape in resnet test case

davidnabergoj · Jun 25, 2024 · 58459cf · 58459cf
1 parent 480c2ff
commit 58459cf
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/normalizing_flows/resnet.py → normalizing_flows/neural_networks/resnet.py b/normalizing_flows/resnet.py → normalizing_flows/neural_networks/resnet.py
@@ -138,7 +138,7 @@ def make_resnet152(image_width, n_outputs):
 
 if __name__ == '__main__':
     n_images = 2
-    image_shape = (3, 270, 270)
+    image_shape = (3, 32, 32)
 
     net = make_resnet18(image_width=image_shape[-1], n_outputs=15)
     y = net(torch.randn(n_images, *image_shape))