From dee93c40e28471c311c1c921debf754de7b691d1 Mon Sep 17 00:00:00 2001 From: Yingda Chen Date: Sat, 22 Oct 2022 16:28:30 +0800 Subject: [PATCH] [to #42322933] force download dataset for portraint enhancement --- tests/trainers/test_image_portrait_enhancement_trainer.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/tests/trainers/test_image_portrait_enhancement_trainer.py b/tests/trainers/test_image_portrait_enhancement_trainer.py index 5c47a59b..123e0098 100644 --- a/tests/trainers/test_image_portrait_enhancement_trainer.py +++ b/tests/trainers/test_image_portrait_enhancement_trainer.py @@ -37,13 +37,13 @@ class TestImagePortraitEnhancementTrainer(unittest.TestCase): namespace='modelscope', subset_name='default', split='test', - download_mode=DownloadMode.REUSE_DATASET_IF_EXISTS)._hf_ds + download_mode=DownloadMode.FORCE_REDOWNLOAD)._hf_ds dataset_val = MsDataset.load( 'image-portrait-enhancement-dataset', namespace='modelscope', subset_name='default', split='test', - download_mode=DownloadMode.REUSE_DATASET_IF_EXISTS)._hf_ds + download_mode=DownloadMode.FORCE_REDOWNLOAD)._hf_ds self.dataset_train = ImagePortraitEnhancementDataset( dataset_train, is_train=True)