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 :)
