Train model parameters and scaler parameters together

2

If your post processing method uses nn.Parameters, you could simply pass them additionally to the optimizer as described here.

1 Like