How to step the optimizer twice inside one training loop?

Hey,

It looks like you need our manual optimization mode in which you can do exactly this!

Let me know if you still have any questions.

Cheers,
Justus