Rasooli
/

KABERT-sentiment-analysis

Model card Files Files and versions

Rasooli commited on about 1 month ago

Commit

f24f534

·

verified ·

1 Parent(s): 4c473b4

Create README.md

Files changed (1) hide show

README.md +6 -0

README.md ADDED Viewed

	@@ -0,0 +1,6 @@

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