For all video prediction models, the batch size was fixed at 2

Models trained with epoch-based schedules were optimized for 300 epochs, while models trained with iterationbased schedules explicitly report the corresponding iteration counts · 2062 · arXiv 6863.1415