from configs import transforms_config from configs.paths_config import dataset_paths DATASETS = { 'ffhq_aging': { 'transforms': transforms_config.AgingTransforms, 'train_source_root': dataset_paths['ffhq'], 'train_target_root': dataset_paths['ffhq'], 'test_source_root': dataset_paths['celeba_test'], 'test_target_root': dataset_paths['celeba_test'], } }