Update tasks/text.py
Browse files- tasks/text.py +1 -1
tasks/text.py
CHANGED
|
@@ -101,7 +101,7 @@ async def evaluate_text(request: TextEvaluationRequest):
|
|
| 101 |
return None
|
| 102 |
|
| 103 |
val_dataset = encode_data(tokenizer, test_dataset['quote'], test_dataset['label'], MAX_LENGTH)
|
| 104 |
-
val_loader = DataLoader(val_dataset, batch_size=
|
| 105 |
|
| 106 |
|
| 107 |
device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
|
|
|
|
| 101 |
return None
|
| 102 |
|
| 103 |
val_dataset = encode_data(tokenizer, test_dataset['quote'], test_dataset['label'], MAX_LENGTH)
|
| 104 |
+
val_loader = DataLoader(val_dataset, batch_size= 16, shuffle=False)
|
| 105 |
|
| 106 |
|
| 107 |
device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
|