Update README.md
Browse files
README.md
CHANGED
|
@@ -121,7 +121,7 @@ max_time = 4.5 # sec
|
|
| 121 |
sense_definitions = list()
|
| 122 |
with torch.no_grad():
|
| 123 |
for i in range(0, len(tokenized_dataset), batch_size):
|
| 124 |
-
batch =
|
| 125 |
|
| 126 |
model_input = dict()
|
| 127 |
for k in ['input_ids', 'attention_mask']:
|
|
|
|
| 121 |
sense_definitions = list()
|
| 122 |
with torch.no_grad():
|
| 123 |
for i in range(0, len(tokenized_dataset), batch_size):
|
| 124 |
+
batch = tokenized_dataset[i:i + batch_size]
|
| 125 |
|
| 126 |
model_input = dict()
|
| 127 |
for k in ['input_ids', 'attention_mask']:
|