nusnlp/greco

how to get the `models/checkpoint.bin` file

Yang-HangWA opened this issue · 0 comments

Thanks for your work on gec. I have a problem about how to get the models/checkpoint.bin file. Do I need to retrain to obtain it, or will an already trained one be provided for testing? The relevant code is as follows.

import torch
from models import GRECO

model = GRECO('microsoft/deberta-v3-large').to(device)
model.load_state_dict(torch.load('models/checkpoint.bin))
model.score(source, hyphoteses)