test with more data

impICNF · Mar 25, 2024 · 5c5ea49 · 5c5ea49
1 parent 392d6d4
commit 5c5ea49
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/test/call_tests.jl b/test/call_tests.jl
@@ -22,8 +22,8 @@
         ]
     end
     omodes = ContinuousNormalizingFlows.Mode[TrainMode(), TestMode()]
-    ndata_ = Int[4]
-    nvars_ = Int[2]
+    ndata_ = Int[2^6]
+    nvars_ = Int[2^3]
     aug_steers = Bool[false, true]
     inplaces = Bool[false, true]
     adb_list = AbstractDifferentiation.AbstractBackend[

diff --git a/test/fit_tests.jl b/test/fit_tests.jl
@@ -22,8 +22,8 @@
         ]
     end
     n_epochs_ = Int[2]
-    ndata_ = Int[4]
-    nvars_ = Int[2]
+    ndata_ = Int[2^6]
+    nvars_ = Int[2^3]
     aug_steers = Bool[false, true]
     inplaces = Bool[false, true]
     adtypes = ADTypes.AbstractADType[