We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 3d0f71f commit 6777498Copy full SHA for 6777498
1 file changed
src/graphnet/training/loss_functions.py
@@ -64,7 +64,10 @@ def _forward(self, prediction: Tensor, target: Tensor) -> Tensor:
64
65
66
class MAELoss(LossFunction):
67
+ """Mean absolute error loss."""
68
+
69
def _forward(self, prediction: Tensor, target: Tensor) -> Tensor:
70
+ """Implement loss calculation."""
71
return torch.mean(torch.abs(prediction - target), dim=-1)
72
73
0 commit comments