how to train the model on CPU ??

[msiraj83]

Currently the code use GPU by default. Can we train it by using CPU??

[arnab39]

I hope you have figured out by now. You can do it using gpu_ids = ''(an empty string). Sorry for not getting back earlier. I'll also try to provide a new version of this code soon.