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'], | |
} | |
} | |