Fix training loop

Fix the l2 regularisation term

Merge request reports

Loading