JxCode

Maybe you should check if there's enough memory you can use ```nvidia-smi``` to check the memory usage and which process is using the GPU  And check the size of...

I'm not the author, but I do trained with my datasets and it worked. Usually, about 8 hours with Nvidia's 1080Ti to get some interesting looking results