Jaren
/

chinese-roberta-wwm-ext-large-NLI

Model card Files Files and versions Community

Jaren commited on Aug 23, 2022

Commit

13b32c9

•

1 Parent(s): c8679a5

Update README.md

Files changed (1) hide show

README.md +2 -1

README.md CHANGED Viewed

@@ -32,4 +32,5 @@ class RobertaClassificationHead(nn.Module):
         x = self.dropout(x)
         x = self.out_proj(x)
         return x
-model = RobertaForSequenceClassification(num_labels)

         x = self.dropout(x)
         x = self.out_proj(x)
         return x
+model = RobertaForSequenceClassification(num_labels)
+model.load_state_dict(torch.load(args.model_save_path+'Roberta_large_model.pt', map_location=device))