jessicayjm
commited on
Commit
•
f798a31
1
Parent(s):
496cffd
update sample code
Browse files
README.md
CHANGED
@@ -50,7 +50,7 @@ class SNN(nn.Module):
|
|
50 |
pred = self.cos(meanPooled1, meanPooled2)
|
51 |
return pred
|
52 |
|
53 |
-
checkpoint_path = '
|
54 |
|
55 |
tokenizer = AutoTokenizer.from_pretrained('sentence-transformers/all-mpnet-base-v2')
|
56 |
model = SNN('sentence-transformers/all-mpnet-base-v2').to('cuda')
|
|
|
50 |
pred = self.cos(meanPooled1, meanPooled2)
|
51 |
return pred
|
52 |
|
53 |
+
checkpoint_path = 'your_path_to/empathy-appraisal-alignment.pt'
|
54 |
|
55 |
tokenizer = AutoTokenizer.from_pretrained('sentence-transformers/all-mpnet-base-v2')
|
56 |
model = SNN('sentence-transformers/all-mpnet-base-v2').to('cuda')
|