hbanduk commited on
Commit
6842466
·
verified ·
1 Parent(s): 9aaf0a5

Update tasks/text.py

Browse files
Files changed (1) hide show
  1. 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= batch_size, shuffle=False)
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")