README.md

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()