- The from_logits=True attribute inform the loss function that the output values generated by the model are not normalized
- In other words, the softmax function has not been applied on them to produce a probability distribution
- Basically we need to softmax and pick max value or the max value from list is the prediction
Keep Learning!!!
No comments:
Post a Comment