Jaren
/

chinese-roberta-wwm-ext-large-NLI

Model card Files Files and versions Community

Jaren commited on Aug 29, 2022

Commit

e250e9a

•

1 Parent(s): 19a8a46

Update README.md

Files changed (1) hide show

README.md +2 -1

README.md CHANGED Viewed

@@ -35,4 +35,5 @@ class RobertaClassificationHead(nn.Module):
         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))```

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