NaN

I mean, the complete Attribute Prediction dataset which contains around 50 categories. Your trained model seems to be trained on only 20 categories.

I am trying to train it for 50 categories. I set the freeze to True and trained for 10 epochs. Then, as per your instruction, I have modified the config...