The issue of mse_loss_weights related to snr

[haohang96]

According to the official implementation of min-snr, in v-parametrization, the mse_loss_weight shall be (Ref to this line: code-url):

$$min(snr, \gamma) / (snr + 1)$$

But in your implementation, it is: (Ref to this line: code-url)

$$min(snr+1, \gamma) / (snr +1)$$

It's because you do add one before mse_loss_weight computation.