search

Nealcly / templateNER

Source code for template-based NER
209 stars 39 forks source link

Output duplication #7

Closed JoyDajunSpaceCraft closed 2 years ago

JoyDajunSpaceCraft commented 3 years ago

Hi, I tried your code and once I use your demo to generate output I expected ["original text","predict entity with prompt"], however it just output ['JapanJapanJapan JapanJapanJapan']. Is that any layer you add to fine-tuning but not shown in code?

Nealcly commented 2 years ago

Could you train more steps? The model is able to generate the reasonable sequence directly.