fix Co-authored-by: ydshieh <ydshieh@users.noreply.github.com>
This commit is contained in:
@@ -304,7 +304,7 @@ class ResNetModelIntegrationTest(unittest.TestCase):
|
||||
expectations = Expectations(
|
||||
{
|
||||
(None, None): [-11.1069, -9.7877, -8.3777],
|
||||
("cuda", 8): [-11.1112, -9.7916, -8.3788],
|
||||
("cuda", 8): [-11.1069, -9.7877, -8.3777],
|
||||
}
|
||||
)
|
||||
expected_slice = torch.tensor(expectations.get_expectation()).to(torch_device)
|
||||
|
||||
Reference in New Issue
Block a user