[to #42322933]支持从dataset json文件中获取参数

* dataset json file add args
This commit is contained in:
feiwu.yfw
2022-08-30 15:15:15 +08:00
parent 745bd5a9e0
commit 2b64cf2bb6
6 changed files with 29 additions and 37 deletions

View File

@@ -36,9 +36,8 @@ class MsDatasetTest(unittest.TestCase):
ms_ds_train = MsDataset.load(
'pets_small',
namespace=DEFAULT_DATASET_NAMESPACE,
split='train',
classes=('1', '2'),
folder_name='Pets')
download_mode=DownloadMode.FORCE_REDOWNLOAD,
split='train')
print(ms_ds_train.config_kwargs)
assert next(iter(ms_ds_train.config_kwargs['split_config'].values()))