Training Prednet for up to t+5 prediction

coxlab / prednet

Code and models accompanying "Deep Predictive Coding Networks for Video Prediction and Unsupervised Learning"

https://arxiv.org/abs/1605.08104

MIT License

759 stars 259 forks source link

Training Prednet for up to t+5 prediction #71

Open Roy-Rupak opened 4 years ago

Roy-Rupak commented 4 years ago

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.