Hugging Face's logo

Jaren
/
chinese-roberta-wwm-ext-large-NLI

Model card Files Community
1
Jaren commited on
Commit
e250e9a
•
1 Parent(s): 19a8a46

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +2 -1
README.md CHANGED
@@ -35,4 +35,5 @@ class RobertaClassificationHead(nn.Module):
35
  x = self.out_proj(x)
36
  return x
37
  model = RobertaForSequenceClassification(num_labels)
38
- model.load_state_dict(torch.load(args.model_save_path+'Roberta_large_model.pt', map_location=device))```
 
 
35
  x = self.out_proj(x)
36
  return x
37
  model = RobertaForSequenceClassification(num_labels)
38
+ model.load_state_dict(torch.load(args.model_save_path+'Roberta_large_model.pt', map_location=device))
39
+ ```