search

NVlabs / NVAE

The Official PyTorch Implementation of "NVAE: A Deep Hierarchical Variational Autoencoder" (NeurIPS 2020 spotlight paper)
https://arxiv.org/abs/2007.03898
Other
999 stars 163 forks source link

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

Open xbowlove opened 11 months ago

xbowlove commented 11 months ago

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 commented 11 months ago

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.