Training Prednet for up to t+5 prediction

[Roy-Rupak]

Hi @bill-lotter
I did the following things:
-Ran the kitti_train.py
-Ran kitti_evaluate.py
It worked just fine. But when I try to run the kitti_extrap_finetune.py , I get the following error:
ValueError: Error when checking input: expected input_1 to have shape (10, 3, 128, 160) but got array with shape (10, 128, 160, 3)
I have not made any changes to kitti_extrap_finetine.py file.
Thanks.