Is it possible to know the loss/parameters in each step (rather than each function call).
Is it possible to know the loss/parameters in each step (rather than each function call).