From 58459cfa6ebe47de684a1a0a2d3381bed4389bbc Mon Sep 17 00:00:00 2001 From: David Nabergoj Date: Tue, 25 Jun 2024 11:46:19 +0200 Subject: [PATCH] Change image shape in resnet test case --- normalizing_flows/{ => neural_networks}/resnet.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) rename normalizing_flows/{ => neural_networks}/resnet.py (99%) diff --git a/normalizing_flows/resnet.py b/normalizing_flows/neural_networks/resnet.py similarity index 99% rename from normalizing_flows/resnet.py rename to normalizing_flows/neural_networks/resnet.py index 0823cd3..9b6c8a9 100644 --- a/normalizing_flows/resnet.py +++ b/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))