About use #15
Description
Hello, will the gradient of loss be calculated when ffl is used as loss? I try to put ffl on cuda, but it is always a scalar, and mse is a tensor with calculated gradient. Here are my changes:
The following are the values of mse and ffl, their forms are different, I don’t know if I’m wrong here.
