ncsnv2
ncsnv2 copied to clipboard
Training on my own semantic dataset
Hi,
I'm new to this area and would like to train/fine tune on my own dataset.
Can you tell me how I can setup my own folder of images for training using main.py?
Thanks
You will need to build a custom data loader in pytorch and replace the data loading pipeline in the provided code. You will need to modify the get_dataset
function.