search

NormXU / ERNIE-Layout-Pytorch

An unofficial Pytorch implementation of ERNIE-Layout which is originally released through PaddleNLP.
http://arxiv.org/abs/2210.06155
MIT License
96 stars 11 forks source link

misleading between num_classes and num_labels #18

Closed huyhoang17 closed 1 year ago

huyhoang17 commented 1 year ago

Hi @NormXU Thanks for your great repo! I used your repo to finetuned for a downstream token-classification task But I think the code had some incorrect block: https://github.com/NormXU/ERNIE-Layout-Pytorch/blob/main/networks/modeling_erine_layout.py#L1050-L1056 I think num_classes and num_label should be the same

NormXU commented 1 year ago

Exactly, thanks for pointing that out. I have fixed it