From f55f274bb445a1c182d29775decef2e53a73f496 Mon Sep 17 00:00:00 2001 From: yanghaitao1 Date: Thu, 17 Sep 2020 10:34:43 -0400 Subject: [PATCH] check list_eval_partition.txt file --- mindspore/dataset/engine/datasets.py | 6 ++++++ 1 file changed, 6 insertions(+) diff --git a/mindspore/dataset/engine/datasets.py b/mindspore/dataset/engine/datasets.py index 319b67f678..be6ef76aa7 100644 --- a/mindspore/dataset/engine/datasets.py +++ b/mindspore/dataset/engine/datasets.py @@ -4874,6 +4874,12 @@ class CelebADataset(MappableDataset): self.shard_id = shard_id self.shuffle_level = shuffle + if usage != "all": + dir = os.path.realpath(self.dataset_dir) + partition_file = os.path.join(dir, "list_eval_partition.txt") + if os.path.exists(partition_file) is False: + raise RuntimeError("Partition file can not be found when usage is not 'all'.") + def get_args(self): args = super().get_args() args["dataset_dir"] = self.dataset_dir