Louis-udm/NER-BERT-CRF

Running code ERROR

sentry0501 opened this issue · 3 comments

sentry0501 commented

I think line 875 at function DataLoader, the collate_fn=pad should be collate_fn=NerDataset.pad ...

stwater20 commented

image

I have the same issue
DengNingyuan commented

me too

DengNingyuan commented

replace by "collate_fn=NerDataset.pad,"
i know ,should be changed to collate_fn=NerDataset.pad,because the author use " collate_fn=NerDataset.pad" when load other data,i guess " collate_fn=pad" may be a mistake.