@rwightman As explained above, during the whole training (maybe optimisation is a better fit here) you only process a single image. You do not train a model, but rather train (optimise) the pixel values of a single image to fit a given loss function.