class MyPath(object): @staticmethod def db_root_dir(database='wjd'): db_names = {'simclr_c10'} assert (database in db_names) if database == 'simclr_c10': return 'D:/wjd/simclr_c10/' else: raise NotImplementedError，这里

如果您想要修改 `db_names`，可以将其改为一个包含多个数据集名称的集合，例如：

class MyPath(object):
    @staticmethod
    def db_root_dir(database='wjd'):
        db_names = {'simclr_c10', 'cifar10', 'imagenet'}
        assert (database in db_names)
        if database == 'simclr_c10':
            return 'D:/wjd/simclr_c10/'
        elif database == 'cifar10':
            return 'D:/datasets/cifar10/'
        elif database == 'imagenet':
            return 'D:/datasets/imagenet/'
        else:
            raise NotImplementedError

这样，您就可以使用 `cifar10` 或 `imagenet` 等其他数据集名称了。当然，您需要将返回的路径修改为您的数据集路径。