from local

This commit is contained in:
dohe0342 2023-05-18 16:18:43 +09:00
parent 6852dc0390
commit 853dbda1ac
2 changed files with 1 additions and 1 deletions

View File

@ -1,7 +1,7 @@
import torch
base_model = torch.load('./d2v-base-T.pt')
bias_model = torch.load('./bitfit_2414_v2/checkpoint-100.pt')
bias_model = torch.load('./bitfit_533_v2/checkpoint-100.pt')
base_model, bias_model = base_model['model'], bias_model['model']