Open Arkkienkeli opened 5 years ago
The same.
I have not tried it myself yet, but I think what you need to do is implement a torch dataset object and register it (see examples /cycada/data). Then modify the training scripts under /scripts and train.
Hello, I would like to train a model on my own data. How it is possible?