Skip to content

Commit 60a74a7

Browse files
committed
fix seed
1 parent 129333d commit 60a74a7

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

tests/test_layers_utils.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -294,6 +294,7 @@ def test_spatial_transformer_dropout(self):
294294
torch.manual_seed(0)
295295
if torch.cuda.is_available():
296296
torch.cuda.manual_seed_all(0)
297+
torch.backends.cudnn.deterministic = True
297298

298299
sample = torch.randn(1, 32, 64, 64).to(torch_device)
299300
spatialTransformerBlock = SpatialTransformer(

0 commit comments

Comments
 (0)