Thanks very much, will give that a try! Just to be clear - this is only done for the training right? Since model.generate() accepts different parameters (no labels) I assume it handles the shifting automatically?