Reduce n_epoch of integration tests

Author: HiroakiMikami

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
 

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

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