fix intermittently failing test

sony · Oct 15, 2024 · 7ebd1b0 · 7ebd1b0
1 parent 2b78d41
commit 7ebd1b0
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tests/pytorch_tests/function_tests/test_hessian_info_calculator.py b/tests/pytorch_tests/function_tests/test_hessian_info_calculator.py
@@ -72,10 +72,10 @@ def forward(self, inp):
 class multiple_outputs_model(torch.nn.Module):
     def __init__(self):
         super(multiple_outputs_model, self).__init__()
-        self.conv1 = Conv2d(3, 3, kernel_size=1, stride=1)
+        self.conv1 = Conv2d(3, 3, kernel_size=3, stride=1, padding=1)
         self.bn1 = BatchNorm2d(3)
         self.relu1 = ReLU()
-        self.conv2 = Conv2d(3, 3, kernel_size=1, stride=1)
+        self.conv2 = Conv2d(3, 3, kernel_size=3, stride=1, padding=1)
         self.bn2 = BatchNorm2d(3)
         self.relu2 = ReLU()
         self.hswish = Hardswish()