[code-quality] Training many large images on a GPU