How can this module be trained?

[guanfuchen]

I want to use pytorch to model prednet, and thanks for your implementing, can this model be trained, and how can I use this repo?

[wmpauli]

Hi, I have been "able" to train the model, but the performance did not exceed the baseline comparison (MSE current frame vs previous frame should be higher than MSE current frame vs. predicted frame). Have you had more luck?