Similar worries to a number of the training steps for these.