You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
It's been too long for me to remember exactly why I implemented things the way I did, but my guess is it's a mistake. The results do still seem to be comparable to the Torch version, but at some point I'll retrain with MSE and see how it affects the output. Good catch!
Hi, I was wondering why you have calculated
sum of squared
in style loss calculations:m_loss = sse(d_mean, s_mean) / batch_size # normalized w.r.t. batch size
While in torch implementation
MSE
is being used, this line:self.mean_criterion = nn.MSECriterion() self.mean_loss = self.mean_criterion:forward(self.input_mean, self.target_mean)
The text was updated successfully, but these errors were encountered: