Loss in on_step_end() callback methods

I need to access the cross entropy loss for the current training step within the on_step_end() method, which won’t accept passing logs to it. How could this be done? Thanks

The loss is not passed along the callbacks right now, so you can’t access it from there.