how to get the `models/checkpoint.bin` file
Yang-HangWA opened this issue · 0 comments
Yang-HangWA commented
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)