mwptoolkit.loss.mse_loss

class mwptoolkit.loss.mse_loss.MSELoss[source]

Bases: AbstractLoss

eval_batch(outputs, target)[source]

calculate loss

Parameters

  • outputs (Tensor) – output distribution of model.

  • target (Tensor) – target distribution.

get_loss()[source]

return loss

Returns

loss (float)