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.
1 Like