how can i use the code on my own dataset. if it's necessary to modify the code carefully myself? about nvae HOT 1 OPEN

well, here is an unofficial solution to this problem. I finally find a way to load your dataset on the code.
first, you can find get_loaders_eval function in dataset.py and add your dataset config in it, mind you are supposed to write your Dataset class carefully and write your transforms for image. then, you should go to AutoEncoder class in model.py to modify the value of self.input_size to your input images' width or height. then you can train your model on your own dataset just need to write the command according to the authors.

from nvae.