deepmodeling / deepmodeling/Uni-Mol
classification
Open
- Dominant language
- Python
- Stars
- 1.2k
- Forks
- 181
- PR merge metrics
- No merged PRs in 30d
Description
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,:]
Contributor guide
No contributing guide indexed for this repository
Assessment
This issue has not been assessed yet.