I am teaching myself the basics of computer vision right now with the imageai library. I have everything set up and working fine on the CPU, but I really want to run the tra