search

mandarjoshi90 / coref

BERT for Coreference Resolution
Apache License 2.0
444 stars 92 forks source link

finetuning on pytorch model #52

Closed HaixiaChai closed 4 years ago

HaixiaChai commented 4 years ago

Hi Mandar,

I wonder if the model can be finetuned using the pytorch version of BERT? I saw that there are tf_checkpoint and init_checkpoint in the config file. Is it possible to modify them in order to support pytorch BERT models.

mandarjoshi90 commented 4 years ago

Hi. Just to make sure I understand the question. Do you mean converting a BERT Pytorch checkpoint to TF, and then finetuning with this repo (using TF)? If so, then yes. See

https://github.com/mandarjoshi90/coref/blob/master/independent.py#L60