I see no reason why you couldn’t add a batch dimension and process multiple images at a time and thus be able to leverage DataParallel. Just make sure you don’t reduce any of the losses across the batch dim.