Spaces:
Running
Running
Update app.py
Browse files
app.py
CHANGED
@@ -31,7 +31,8 @@ def compute_model_card_evaluation_results(tokenizer, model_checkpoint, raw_datas
|
|
31 |
tokenized_datasets = raw_datasets.map(preprocess_function, batched=True)
|
32 |
model = AutoModelForSequenceClassification.from_pretrained(model_checkpoint, num_labels=2)
|
33 |
|
34 |
-
batch_size = 16
|
|
|
35 |
args = TrainingArguments(
|
36 |
"test-glue",
|
37 |
eval_strategy = "epoch",
|
|
|
31 |
tokenized_datasets = raw_datasets.map(preprocess_function, batched=True)
|
32 |
model = AutoModelForSequenceClassification.from_pretrained(model_checkpoint, num_labels=2)
|
33 |
|
34 |
+
#batch_size = 16
|
35 |
+
batch_size = 8
|
36 |
args = TrainingArguments(
|
37 |
"test-glue",
|
38 |
eval_strategy = "epoch",
|