Tips for training ML models on image data