Skip to content

GitLab

Unverified Commit e2e323cc authored by Ashish Farmer's avatar Ashish Farmer Committed by GitHub
Browse files

use deterministic torch for test_linear_transformation (#3099)

parent 7f1a05a3
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 0 deletions
test/test_transforms_tensor.py
View file @ e2e323cc
...@@ -452,6 +452,7 @@ class Tester(TransformsTester): ...@@ -452,6 +452,7 @@ class Tester(TransformsTester):
# We skip some tests from _test_transform_vs_scripted_on_batch as # We skip some tests from _test_transform_vs_scripted_on_batch as
# results for scripted and non-scripted transformations are not exactly the same # results for scripted and non-scripted transformations are not exactly the same
torch.manual_seed(12) torch.manual_seed(12)
torch.set_deterministic(True)
transformed_batch = fn(batch_tensors) transformed_batch = fn(batch_tensors)
torch.manual_seed(12) torch.manual_seed(12)
s_transformed_batch = scripted_fn(batch_tensors) s_transformed_batch = scripted_fn(batch_tensors)
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment