Skip to content

Commit 9ae9be2

Browse files
clee2000Peter Y. Yeh
authored andcommitted
Mark test_xavier_uniform as slow (pytorch#124801)
takes 17+ minutes, sometimes 30+ min Pull Request resolved: pytorch#124801 Approved by: https://github.com/huydhn
1 parent be300be commit 9ae9be2

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

test/nn/test_init.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -14,6 +14,7 @@
1414
run_tests,
1515
skipIfNoLapack,
1616
skipIfTorchDynamo,
17+
slowTest,
1718
TEST_SCIPY,
1819
TestCase,
1920
)
@@ -318,6 +319,7 @@ def test_xavier_normal_errors_on_inputs_smaller_than_2d(self):
318319
init.xavier_normal_(tensor)
319320

320321
@unittest.skipIf(not TEST_SCIPY, "Scipy not found.")
322+
@slowTest
321323
def test_xavier_uniform(self):
322324
for use_gain in [True, False]:
323325
for dims in [2, 4]:

0 commit comments

Comments
 (0)