commit 0b8b9cf9282258ca0f114a35666b979484126d53
author SsnL <tongzhou.wang.1994@gmail.com> Tue Oct 24 13:34:36 2017 -0400
committer Adam Paszke <adam.paszke@gmail.com> Tue Oct 24 20:30:49 2017 +0200
tree ac0f8d33706a26f73e75bf26e3803c02223c0f18
parent 5691b0b8d2d184b961b93f6d27770db78ab51e22

update

test/test_legacy_nn.py

diff --git a/test/test_legacy_nn.py b/test/test_legacy_nn.py
index 535ba13..5dbf6d8 100644
--- a/test/test_legacy_nn.py
+++ b/test/test_legacy_nn.py
@@ -486,14 +486,14 @@
                   input_size=(1, 2, 4, 4, 4)),
     OldModuleTest(nn.VolumetricMaxPooling,
                   constructor_args=(2, 2, 2),
-                  input=(torch.randn(2, 3, 5, 5, 5) * 1000)),
+                  input_fn=lambda: (torch.randn(2, 3, 5, 5, 5) * 1000)),
     OldModuleTest(nn.VolumetricMaxPooling,
                   constructor_args=(2, 2, 2, 2, 2, 2),
-                  input=(torch.randn(2, 3, 5, 5, 5) * 1000),
+                  input_fn=lambda: (torch.randn(2, 3, 5, 5, 5) * 1000),
                   desc='stride'),
     OldModuleTest(nn.VolumetricMaxPooling,
                   constructor_args=(2, 2, 2, 2, 2, 2, 1, 1, 1),
-                  input=(torch.randn(2, 3, 5, 5, 5) * 1000),
+                  input_fn=lambda: (torch.randn(2, 3, 5, 5, 5) * 1000),
                   desc='stride_padding'),
     OldModuleTest(nn.VolumetricReplicationPadding,
                   constructor_args=(1, 2, 3, 4, 5, 6),