Thank you for your interesting work, and replying for my previous question.

I have an another question regarding computing prediction probability distribution of a given data x.

When I run the eval_prob_adaptive.py and get a array of losses with respect to each class prompt, I found that the losses are very close to each other.
That is, when I softmax the array, the probability for each class is close to 1/N which may not be desired.

I found that similar issue had been raised before. (#11 (comment))

Could you check about this issue?
Thank you in advance.