Browse Source

[to #42322933] force download dataset for portraint enhancement

master
Yingda Chen 2 years ago
parent
commit
dee93c40e2
1 changed files with 2 additions and 2 deletions
  1. +2
    -2
      tests/trainers/test_image_portrait_enhancement_trainer.py

+ 2
- 2
tests/trainers/test_image_portrait_enhancement_trainer.py View File

@@ -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)


Loading…
Cancel
Save