removing totalDuration because it inexplicably slows down the training process significantly