How to call ELBO loss function consecutively?

Hi all,

I would like to ask if it is possible to call a loss function consecutively multiple times and with different parameters?

This image illustrates what I am trying to do:

Need to run the ELBO in step 1 for 400 epochs and the ELBO in step 2 for 600 epochs. Note: Both ELBOs have different inputs and I think that this is supposed to be done with a single instance of a model. In addition, I don’t expect people to go and dig into these models, but if you could explain how to do this using your own examples that would be great!

Looking forward to your answer!