Merge e2d8820 into 908ec57

pytorch · Feb 3, 2021 · c1d9280 · c1d9280
2 parents 908ec57 + e2d8820
commit c1d9280
Show file tree

Hide file tree

Showing 3 changed files with 65 additions and 208 deletions.
diff --git a/test/optim/test_initializers.py b/test/optim/test_initializers.py
@@ -159,7 +159,7 @@ def test_gen_batch_initial_conditions(self):
                             self.assertEqual(raw_samps.shape, expected_raw_samps_shape)
 
     def test_gen_batch_initial_conditions_highdim(self):
-        d = 120
+        d = 2200  # 2200 * 10 (q) > 21201 (sobol max dim)
         bounds = torch.stack([torch.zeros(d), torch.ones(d)])
         for dtype in (torch.float, torch.double):
             bounds = bounds.to(device=self.device, dtype=dtype)