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