From 8267fe3602e2448e74f61158f7bc1e60344c6fc7 Mon Sep 17 00:00:00 2001 From: gpucce Date: Thu, 16 Feb 2023 05:59:58 +0900 Subject: [PATCH 1/2] enable eval test --- src/training/data.py | 1 + tests/test_training_simple.py | 2 ++ 2 files changed, 3 insertions(+) diff --git a/src/training/data.py b/src/training/data.py index 8eee19f85..c7248307e 100644 --- a/src/training/data.py +++ b/src/training/data.py @@ -483,6 +483,7 @@ def __getitem__(self, idx): def get_synthetic_dataset(args, preprocess_fn, is_train, epoch=0, tokenizer=None): image_size = preprocess_fn.transforms[0].size + image_size = image_size if isinstance(image_size, tuple) else (image_size, image_size) dataset = SyntheticDataset( transform=preprocess_fn, image_size=image_size, dataset_size=args.train_num_samples, tokenizer=tokenizer) num_samples = len(dataset) diff --git a/tests/test_training_simple.py b/tests/test_training_simple.py index 70998d592..76d0289d0 100644 --- a/tests/test_training_simple.py +++ b/tests/test_training_simple.py @@ -21,6 +21,8 @@ def test_training(): '--zeroshot-frequency', '1', '--dataset-type', "synthetic", '--train-num-samples', '16', + '--val-data', "synthetic", + '--val-num-samples', '16', '--warmup', '1', '--batch-size', '4', '--lr', '1e-3', From 05a701c53ccfbfa2d47026b92cddac8b476511de Mon Sep 17 00:00:00 2001 From: gpucce Date: Thu, 16 Feb 2023 06:13:18 +0900 Subject: [PATCH 2/2] add eval to coca --- tests/test_training_simple.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/tests/test_training_simple.py b/tests/test_training_simple.py index 76d0289d0..828c5fa21 100644 --- a/tests/test_training_simple.py +++ b/tests/test_training_simple.py @@ -39,6 +39,8 @@ def test_training_coca(): '--zeroshot-frequency', '1', '--dataset-type', "synthetic", '--train-num-samples', '16', + '--val-data', 'synthetic', + '--val-num-samples', '16', '--warmup', '1', '--batch-size', '4', '--lr', '1e-3',