Browse Source

[to #42322933]add subset_name when loading dataset (NAFNet image denoising)

Link: https://code.alibaba-inc.com/Ali-MaaS/MaaS-lib/codereview/10427797
master
huizheng.hz yingda.chen 3 years ago
parent
commit
c0b546a96e
1 changed files with 2 additions and 0 deletions
  1. +2
    -0
      tests/trainers/test_image_denoise_trainer.py

+ 2
- 0
tests/trainers/test_image_denoise_trainer.py View File

@@ -33,11 +33,13 @@ class ImageDenoiseTrainerTest(unittest.TestCase):
dataset_train = MsDataset.load( dataset_train = MsDataset.load(
'SIDD', 'SIDD',
namespace='huizheng', namespace='huizheng',
subset_name='default',
split='validation', split='validation',
download_mode=DownloadMode.REUSE_DATASET_IF_EXISTS)._hf_ds download_mode=DownloadMode.REUSE_DATASET_IF_EXISTS)._hf_ds
dataset_val = MsDataset.load( dataset_val = MsDataset.load(
'SIDD', 'SIDD',
namespace='huizheng', namespace='huizheng',
subset_name='default',
split='test', split='test',
download_mode=DownloadMode.REUSE_DATASET_IF_EXISTS)._hf_ds download_mode=DownloadMode.REUSE_DATASET_IF_EXISTS)._hf_ds
self.dataset_train = SiddImageDenoisingDataset( self.dataset_train = SiddImageDenoisingDataset(


Loading…
Cancel
Save