KoichiYasuoka
commited on
Commit
•
a7a424a
1
Parent(s):
3e182db
with torch.nograd
Browse files
README.md
CHANGED
@@ -86,8 +86,9 @@ class TransformersSlowUD(object):
|
|
86 |
q=[self.tokenizer.cls_token_id]+t+[self.tokenizer.sep_token_id]
|
87 |
c.append([q]+v[0:i]+[[self.tokenizer.mask_token_id]]+v[i+1:]+[[q[-1]]])
|
88 |
b=[[len(sum(x[0:j+1],[])) for j in range(len(x))] for x in c]
|
89 |
-
|
90 |
-
|
|
|
91 |
s,e=d.start_logits.tolist(),d.end_logits.tolist()
|
92 |
for i in range(n):
|
93 |
for j in range(n):
|
|
|
86 |
q=[self.tokenizer.cls_token_id]+t+[self.tokenizer.sep_token_id]
|
87 |
c.append([q]+v[0:i]+[[self.tokenizer.mask_token_id]]+v[i+1:]+[[q[-1]]])
|
88 |
b=[[len(sum(x[0:j+1],[])) for j in range(len(x))] for x in c]
|
89 |
+
with torch.no_grad():
|
90 |
+
d=self.model(input_ids=torch.tensor([sum(x,[]) for x in c]),
|
91 |
+
token_type_ids=torch.tensor([[0]*x[0]+[1]*(x[-1]-x[0]) for x in b]))
|
92 |
s,e=d.start_logits.tolist(),d.end_logits.tolist()
|
93 |
for i in range(n):
|
94 |
for j in range(n):
|