Custom dataset

[DianeBouchacourt]

Hi,

Thank you very much for setting this library ! If I am correct, one can use a custom dataset by creating the corresponding data_loader, and updating accordingly the function:

disentanglement-pytorch/common/data_loader.py

Line 182 in 2626252

def _get_dataloader_with_labels(name, dset_dir, batch_size, seed, num_workers, image_size, include_labels, pin_memory,

This will run with the training and evaluation metrics code? There is nothing specific to dSprites and CelebA?

[amir-abdi]

Hi,
I don't think I have made much assumptions on dSprites and/or CelebA datasets. However, looking at the code, I hope I had implemented the data loader following a more object oriented design so that you can create your own dataloader for your own dataset without having to deal with the if/else scenarios for dSprites and CelebA.

Anyhow, to answer your question: yes, you can add your own dataset following what I have done for celebA and dSprites.