Update app.py

app.py

@@ -30,6 +30,7 @@ def compute_metrics(eval_pred):
 def compute_model_card_evaluation_results(tokenizer, model_checkpoint, raw_datasets, metric):
     tokenized_datasets = raw_datasets.map(preprocess_function, batched=True)
     model = AutoModelForSequenceClassification.from_pretrained(model_checkpoint, num_labels=2)
+    """
     batch_size = 16
     args = TrainingArguments(
         "test-glue",
@@ -45,7 +46,22 @@ def compute_model_card_evaluation_results(tokenizer, model_checkpoint, raw_datas
         metric_for_best_model="accuracy",
         report_to="none"
         )
-
+    """
+    batch_size = 1024
+    args = TrainingArguments(
+        "test-glue",
+        eval_strategy = "epoch",
+        learning_rate=0.0007199555649276667,
+        seed=33,
+        lr_scheduler_type="linear",
+        per_device_train_batch_size=batch_size,
+        per_device_eval_batch_size=batch_size,
+        num_train_epochs=7,
+        weight_decay=0.01,
+        load_best_model_at_end=False,
+        metric_for_best_model="accuracy",
+        report_to="none"
+        )
     trainer = Trainer(
         model,
         args,