google-research/electra

Tagging Task Segment ids

kamalkraj opened this issue · 0 comments

kamalkraj commented

Why Tagging Task segment ids are ones instead of zeros?

electra/finetune/tagging/tagging_tasks.py

Line 144 in 8a46635

segment_ids = pad([1] * len(input_ids))

Tagging task only contains the first segment and it should be zeros, right?

@clarkkev