Reduce n_epoch of integration tests

HiroakiMikami · Oct 25, 2020 · 628d567 · 628d567
1 parent b982c21
commit 628d567
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 6 deletions.
diff --git a/test_integration/test_csg_by_pbe_with_repl.py b/test_integration/test_csg_by_pbe_with_repl.py
@@ -240,8 +240,8 @@ def pretrain(self, output_dir):
                 ])),
                 None, "score",
                 collate_fn,
-                1, Epoch(100), evaluation_interval=Epoch(10),
-                snapshot_interval=Epoch(100)
+                1, Epoch(50), evaluation_interval=Epoch(50),
+                snapshot_interval=Epoch(50)
             )
         return encoder, train_dataset
 

diff --git a/test_integration/test_nl2code.py b/test_integration/test_nl2code.py
@@ -155,8 +155,8 @@ def train(self, output_dir):
                 ),
                 "accuracy@5",
                 lambda x: collate(transform(x)),
-                1, Epoch(100), evaluation_interval=Epoch(100),
-                snapshot_interval=Epoch(100),
+                1, Epoch(50), evaluation_interval=Epoch(50),
+                snapshot_interval=Epoch(50),
                 threshold=1.0
             )
         return qencoder, aencoder

diff --git a/test_integration/test_treegen.py b/test_integration/test_treegen.py
@@ -171,8 +171,8 @@ def train(self, output_dir):
                 ),
                 "accuracy@5",
                 lambda x: collate(transform(x)),
-                1, Epoch(100), evaluation_interval=Epoch(100),
-                snapshot_interval=Epoch(100),
+                1, Epoch(50), evaluation_interval=Epoch(50),
+                snapshot_interval=Epoch(50),
                 threshold=1.0
             )
         return encoder