Flux train! no longer works

I have a neural network which has two parts. one encoder and the other main network. Due to recent changes in Flux.jl API, my code no longer works. I keep some of weights of main network and train them beside encoder. how to do it in newer flux.jl? do I have to pass two networks as model into loss(model, x, y)?