Thanks for the link, but isn’t this breaking the graph? I ask you because I read this Calculating loss with numpy function.