Error when running kitti_extrap_finetune.py

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

Error when running kitti_extrap_finetune.py #77

Open lwtt opened 4 years ago

lwtt commented 4 years ago

When I reproduced your code, the following error appeared, I did not modify anything. Hope you can answer me, thank you

--->predictions = prednet(inputs) ValueError: You called set_weights(weights) on layer "pred_net_1" with a weight list of length 46, but the layer was expecting 47 weights. Provided weights: [array([[[[ 1.83418002e-02, -6.57398626e-02, -1.06...

Siamul commented 2 years ago

I found that downgrading keras and tensorflow fixes this. pip install tensorflow==1.15.5 pip install tensorflow-gpu==1.15.5 pip install keras==2.2.4