generation/data/base_dataset.py
separate_vae/data/base_dataset.py

generation/data/pickle_dataset.py
separate_vae/data/pickle_dataset.py

generation/data/custom_dataset_data_loader.py
separate_vae/data/custom_dataset_data_loader.py

generation/data/base_data_loader.py
separate_vae/data/base_data_loader.py