Unaveraged MSE loss criterion

Thank you! I thought reduction= ‘none’ was the default, my mistake.