Can you provide the complete training code, including the optical flow branch?

[lcf000000]

I don't know how to train the optical flow branch, including how to switch to another branch after training one branch reach the threshold. How should I set it ?