mlcolvar.core.loss.MSELoss¶

class mlcolvar.core.loss.MSELoss(*args: Any, **kwargs: Any)[source]¶

Bases: Module

(Weighted) Mean Square Error

__init__(*args: Any, **kwargs: Any) → None¶: Initialize internal Module state, shared by both nn.Module and ScriptModule.

Methods

forward(input, target[, weights])

Compute the value of the loss function.

forward(input: Tensor, target: Tensor, weights: Tensor | None = None) → Tensor[source]¶: Compute the value of the loss function.

Attributes

T_destination

call_super_init

dump_patches

training