Filter size runtime error when stack_sizes and R_stack_sizes are set with different channels

coxlab / prednet

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

https://arxiv.org/abs/1605.08104

MIT License

765 stars 260 forks source link

Closed RoshanRane closed 5 years ago

RoshanRane commented 5 years ago

The bug is caused due to line 305 in the build() method of PredNet class. The fix is - nb_channels = 2 * self.stack_sizes[l]

bill-lotter commented 5 years ago

Good catch, thanks!