Instructions on how to train the models on additional data?

[arikanev]

Thanks for this script! It is fantastic, but I would like to expand the distribution on which it was trained on - any pointers to where I might be able to get that code??

Thanks!