ksatzke commited on
Commit
7d5bc47
·
verified ·
1 Parent(s): 117043a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
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",