Skip to content

Update how loss function work

Cyril Moineau requested to merge bindLoss into dev

Context

Currently, Loss function only compute loss but do not create the gradient of the graph output node.

This MR aims at fixing this issue following the solution described in: #1 (closed)

Modified files

  • MSE function now set the mGrad of the pred Tensor.
  • Bind loss function MSE
Edited by Cyril Moineau

Merge request reports

Loading