HINT: Use smaller image size for testing, will consume less of your free GPU time!