AdamCodd
/

distilroberta-query-wellformedness

Text Classification

Model card Files Files and versions Community

AdamCodd commited on Oct 19, 2023

Commit

7e4ae0f

·

1 Parent(s): 071eeeb

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -58,7 +58,7 @@ class RegressionModel(torch.nn.Module):
 regression_model = RegressionModel()
 # Do not forget to set the correct path to load the regression head
-regression_model.regression_head.load_state_dict(torch.load(r"path_to_the_regression_head.pth"))
 regression_model.eval()
 # Examples
 sentences = [

 regression_model = RegressionModel()
 # Do not forget to set the correct path to load the regression head
+regression_model.regression_head.load_state_dict(torch.load("path_to_the_regression_head.pth"))
 regression_model.eval()
 # Examples
 sentences = [