saymrwulf/onnxruntime
1
0
Fork 0
mirror of https://github.com/saymrwulf/onnxruntime.git synced 2026-07-16 18:31:27 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Pinning manual seed (#12714)

Browse source
This commit is contained in:
abhi-ort 2022-08-25 10:09:02 -07:00 committed by GitHub
parent e85dce8cea
commit ebff15d743
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 10 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

10
orttraining/orttraining/eager/test/ort_ops.py
View file

@ -221,6 +221,11 @@ class OrtOpTests(unittest.TestCase):
]
)
def test_softmax_grad(self, input_shape, dim):
# The 1% tolerance used by this test is not working for any random inputs
# and on the other hand it is tough to come up with some tolerance value
# that works for any random input values. So, pin the seed value so that the
# random inputs used by this test are always the same.
torch.manual_seed(5)
device = self.get_device()
cpu_tensor = torch.nn.Parameter(torch.rand(input_shape))
ort_tensor = torch.nn.Parameter(cpu_tensor.detach().clone().to(device))
@ -243,6 +248,11 @@ class OrtOpTests(unittest.TestCase):
]
)
def test_logsoftmax_grad(self, input_shape, dim):
# The 5% tolerance used by this test is not working for any random inputs
# and on the other hand it is tough to come up with some tolerance value
# that works for any random input values. So, pin the seed value so that the
# random inputs used by this test are always the same.
torch.manual_seed(5)
device = self.get_device()
cpu_tensor = torch.nn.Parameter(torch.rand(input_shape))
ort_tensor = torch.nn.Parameter(cpu_tensor.detach().clone().to(device))