Loss in on_step_end() callback methods
I need to access the cross entropy loss for the current training step within theon_step_end()method, which won’t accept passinglogsto it. How could this be done? Thanks
2021-02-19T20:51:16Z
The loss is not passed along the callbacks right now, so you can’t access it from there.