jhoffman / cycada_release

Code to accompany ICML 2018 paper
BSD 2-Clause "Simplified" License
558 stars 126 forks source link

How to train a model on my own data? #15

Open Arkkienkeli opened 5 years ago

Arkkienkeli commented 5 years ago

Hello, I would like to train a model on my own data. How it is possible?

happsky commented 5 years ago

The same.

hannesliik commented 5 years ago

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.