DEV Community

Discussion on: PyTorch Hello World

Collapse
 
ivej profile image
IveJ

The only difference is that PyTorch's MSELoss function doesn't have the extra division by 2, so in the code below, I've adjusted dc_da_l2 = 2 * (a_l2-1) to match what PyTorch does

=> And so you dont exponent by 2!?

Collapse
 
nestedsoftware profile image
Nested Software • Edited

Thanks for the question - this is the derivative of the loss function (with respect to the output).