Hello,
I wanted to ask whether during accuracy update as well softmax function is applied internally just like in compute_loss function?
Thank you.
Regards,
Akhilesh Chandgude
Update to state of train_accuracy is tied to output predictions of the model. See update_state, reset_state and result methods here