I\'m using tfa.losses.WeightedKappa() for multi-class classification problem. Loss goes NaN after few epochs. What can be the reason?
This is my code: