use deterministic torch for test_linear_transformation (#3099)

e2e323cc · Ashish Farmer · GitHub · 7f1a05a3 · e2e323cc
Unverified Commit e2e323cc authored Dec 03, 2020 by Ashish Farmer Committed by GitHub Dec 03, 2020
Show whitespace changes
Inline Side-by-side

Showing with 1 addition and 0 deletions

test/test_transforms_tensor.py test/test_transforms_tensor.py +1 -0

No files found.
--- a/test/test_transforms_tensor.py
+++ b/test/test_transforms_tensor.py
@@ -452,6 +452,7 @@ class Tester(TransformsTester):
        # We skip some tests from _test_transform_vs_scripted_on_batch as
        # results for scripted and non-scripted transformations are not exactly the same
        torch.manual_seed(12)
+        torch.set_deterministic(True)
        transformed_batch = fn(batch_tensors)
        torch.manual_seed(12)
        s_transformed_batch = scripted_fn(batch_tensors)