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()