Bug report. I think we need MSELoss.

IrisRainbowNeko / HCP-Diffusion

A universal Stable-Diffusion toolbox

Apache License 2.0

899 stars 75 forks source link

Bug report. I think we need MSELoss. #38

Closed iamwangyabin closed 11 months ago

iamwangyabin commented 1 year ago

Please do no remove mse_loss.py. I noticed that the file "mse_loss.py" has been removed, specifically the MSELoss module. However, this removal has caused some unexpected errors in the current version.

IrisRainbowNeko commented 11 months ago

The new version has support for using torch's native torch.nn.MSELoss directly, and the redundant custom MSELoss is removed. All similar Losses can be used directly without the need for additional packaging.