classification

[alanianeselli]

At line 348 of models.unimol2.py, why are you passing x[:, 1:, :] to the classification head? If you look at the ClassificationHead class (line 417) you can see that it performs classification on the CLS token, therefore x[:,0,:]

[robotcator]

Thank you for your feedback! This mistake was caused by reusing the old version code during the release process, and we’ve now submitted a new PR to fix it.