Adding additional variable to optimizer params

[Kai_Wang]

Did you solve this issue? I’m having a similar problem