From 628d567f0f0db9e7a7a737e756a2b114fbf9ec36 Mon Sep 17 00:00:00 2001 From: Hiroaki Mikami Date: Sun, 25 Oct 2020 13:14:19 +0900 Subject: [PATCH] Reduce n_epoch of integration tests --- test_integration/test_csg_by_pbe_with_repl.py | 4 ++-- test_integration/test_nl2code.py | 4 ++-- test_integration/test_treegen.py | 4 ++-- 3 files changed, 6 insertions(+), 6 deletions(-) diff --git a/test_integration/test_csg_by_pbe_with_repl.py b/test_integration/test_csg_by_pbe_with_repl.py index 30702cc..318caa3 100644 --- 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 index 3165b0b..9d6f327 100644 --- 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 index d3105ea..a1a0b5d 100644 --- 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