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)