File size: 190 Bytes
f24f534 |
1 2 3 4 5 6 7 |
import torch
from kabert import KABERT
model = KABERT("bert-base-multilingual-cased", num_labels=3)
model.load_state_dict(torch.load("kabert_gpu_best.pt", map_location="cpu"))
model.eval()
|