mandarjoshi90 / coref

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

training with my dataset #32

Closed mujizi closed 4 years ago

mujizi commented 4 years ago

hello author: I want to use this res code to train my model with another dataset. But i find original spanbert is a pytorch.bin file. config file is ckpt model loader. I can't load spanbert.

best wish

mandarjoshi90 commented 4 years ago

Please take a look at this -- https://github.com/mandarjoshi90/coref/blob/master/pytorch_to_tf.py#L38