From a1738690c9f086a09831da059a5f7a6bb1736ebb Mon Sep 17 00:00:00 2001 From: "huizheng.hz" Date: Tue, 25 Oct 2022 10:08:57 +0800 Subject: [PATCH] [to #42322933]test_image_denoise_trainer Link: https://code.alibaba-inc.com/Ali-MaaS/MaaS-lib/codereview/10465138 --- tests/trainers/test_image_denoise_trainer.py | 10 +++++----- 1 file changed, 5 insertions(+), 5 deletions(-) diff --git a/tests/trainers/test_image_denoise_trainer.py b/tests/trainers/test_image_denoise_trainer.py index c4abca6a..b742dcae 100644 --- a/tests/trainers/test_image_denoise_trainer.py +++ b/tests/trainers/test_image_denoise_trainer.py @@ -34,14 +34,14 @@ class ImageDenoiseTrainerTest(unittest.TestCase): 'SIDD', namespace='huizheng', subset_name='default', - split='validation', - download_mode=DownloadMode.REUSE_DATASET_IF_EXISTS)._hf_ds + split='test', + download_mode=DownloadMode.FORCE_REDOWNLOAD)._hf_ds dataset_val = MsDataset.load( 'SIDD', namespace='huizheng', subset_name='default', split='test', - download_mode=DownloadMode.REUSE_DATASET_IF_EXISTS)._hf_ds + download_mode=DownloadMode.FORCE_REDOWNLOAD)._hf_ds self.dataset_train = SiddImageDenoisingDataset( dataset_train, self.config.dataset, is_train=True) self.dataset_val = SiddImageDenoisingDataset( @@ -51,7 +51,7 @@ class ImageDenoiseTrainerTest(unittest.TestCase): shutil.rmtree(self.tmp_dir, ignore_errors=True) super().tearDown() - @unittest.skipUnless(test_level() >= 2, 'skip test in current test level') + @unittest.skipUnless(test_level() >= 0, 'skip test in current test level') def test_trainer(self): kwargs = dict( model=self.model_id, @@ -65,7 +65,7 @@ class ImageDenoiseTrainerTest(unittest.TestCase): for i in range(2): self.assertIn(f'epoch_{i+1}.pth', results_files) - @unittest.skipUnless(test_level() >= 2, 'skip test in current test level') + @unittest.skipUnless(test_level() >= 1, 'skip test in current test level') def test_trainer_with_model_and_args(self): model = NAFNetForImageDenoise.from_pretrained(self.cache_path) kwargs = dict(