Use multiple checkpoints for neural network input