Skip to content

Commit 7ba5e7c

Browse files
authored
fix VolumetricMaxPooling test instability (pytorch#952)
1 parent 9b626a8 commit 7ba5e7c

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

test/test_legacy_nn.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -483,7 +483,7 @@ def _do_test(self, test_case, module, input):
483483
input_size=(1, 2, 4, 4, 4)),
484484
OldModuleTest(nn.VolumetricMaxPooling,
485485
(2, 2, 2),
486-
input_size=(2, 3, 5, 5, 5)),
486+
input=(torch.randn(2, 3, 5, 5, 5) * 1000)),
487487
OldModuleTest(nn.VolumetricMaxPooling,
488488
(2, 2, 2, 2, 2, 2),
489489
input_size=(2, 3, 5, 5, 5),

0 commit comments

Comments
 (0)