From 7de172f8466d1f8de62601fb7d5ebd9dcd9a681c Mon Sep 17 00:00:00 2001 From: Liv Gorton Date: Sat, 18 May 2024 18:05:44 -0700 Subject: [PATCH] fix alpha value for LocalResponseNorm0 and LocalResponseNorm1 --- lucent/modelzoo/inceptionv1/InceptionV1.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/lucent/modelzoo/inceptionv1/InceptionV1.py b/lucent/modelzoo/inceptionv1/InceptionV1.py index 4c21e58..6f8543a 100644 --- a/lucent/modelzoo/inceptionv1/InceptionV1.py +++ b/lucent/modelzoo/inceptionv1/InceptionV1.py @@ -201,13 +201,13 @@ def forward(self, x): conv2d0 = self.conv2d0(conv2d0_pre_relu_conv) maxpool0_pad = F.pad(conv2d0, (0, 1, 0, 1), value=float('-inf')) maxpool0 = self.maxpool0(maxpool0_pad, kernel_size=(3, 3), stride=(2, 2), padding=0, ceil_mode=False) - localresponsenorm0 = F.local_response_norm(maxpool0, size=9, alpha=9.99999974738e-05, beta=0.5, k=1) + localresponsenorm0 = F.local_response_norm(maxpool0, size=10, alpha=10 * 0.00009999999747378752, beta=0.5, k=2) conv2d1_pre_relu_conv = self.conv2d1_pre_relu_conv(localresponsenorm0) conv2d1 = self.conv2d1(conv2d1_pre_relu_conv) conv2d2_pre_relu_conv_pad = F.pad(conv2d1, (1, 1, 1, 1)) conv2d2_pre_relu_conv = self.conv2d2_pre_relu_conv(conv2d2_pre_relu_conv_pad) conv2d2 = self.conv2d2(conv2d2_pre_relu_conv) - localresponsenorm1 = F.local_response_norm(conv2d2, size=9, alpha=9.99999974738e-05, beta=0.5, k=1) + localresponsenorm1 = F.local_response_norm(conv2d2, size=10, alpha=10 * 0.00009999999747378752, beta=0.5, k=2) maxpool1_pad = F.pad(localresponsenorm1, (0, 1, 0, 1), value=float('-inf')) maxpool1 = self.maxpool1(maxpool1_pad, kernel_size=(3, 3), stride=(2, 2), padding=0, ceil_mode=False) mixed3a_1x1_pre_relu_conv = self.mixed3a_1x1_pre_relu_conv(maxpool1)