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

[xbowlove]

I'd like to train the model on my own dataset.but i don't know how to use the code to train it. do i need to read the code carefully to modify the code to train it. or if i can just use the config file or command without to read the detailed code? time is really tight so I perfer if someone can help me answer this question.

[xbowlove]

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.