which checkpoint of the trained weights should I use?

lxucs / coref-hoi

PyTorch implementation of the end-to-end coreference resolution model with different higher-order inference methods.

Apache License 2.0

59 stars 19 forks source link

which checkpoint of the trained weights should I use? #12

Closed world2vec closed 3 years ago

world2vec commented 3 years ago

Hi lxucs, There are 2 checkpoint of the trained weights, which one is the one used in your paper? Thanks

Below is an example:

train_spanbert_large_ml0_cm_fn1000_max_dloss/model_May14_05-15-38_63000.bin
train_spanbert_large_ml0_cm_fn1000_max_dloss/model_May22_23-31-16_66000.bin

lxucs commented 3 years ago

Hi, May14_05-15-38_63000 has 80.19 F1 and May22_23-31-16_66000 has 80.13 F1.