pytorch-lightning

120c87f8 - Include the training mode in the ModelSummary (#19468)