Hi @KFrank, if I want to use the MSE loss instead of cross_entropy then how I have to write the code?
More specifically, are there any variants for MSE loss. For example, we are using cross_entrpoy for binary-class classification and logits for multi-class classification.
So, MSE loss has 2 types (one for binary class nad another for multi-class)?
Sorry, for this kind of silly questions!