From 7ebd1b02accab3d8c1e7ee0e8c9d884ed930d157 Mon Sep 17 00:00:00 2001 From: irenab Date: Mon, 14 Oct 2024 22:07:59 +0300 Subject: [PATCH] fix intermittently failing test --- .../function_tests/test_hessian_info_calculator.py | 4 ++-- 1 file changed, 2 insertions(+), 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 index 8d9f7318c..bf8a39067 100644 --- 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()