This is super cool! Although I notice it uses CPU to train, is there an option to use GPU instead?
Thank you for checking out the project! An option to use GPU instead of CPU for training is a great idea, I might look into implementing it soon. Thanks for the feedback :)