grpathak22 commited on
Commit
36fa541
1 Parent(s): e3660e1

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -42,8 +42,8 @@ training_args = TrainingArguments(
42
  output_dir="./results",
43
  evaluation_strategy="epoch",
44
  learning_rate=2e-5,
45
- per_device_train_batch_size=4,
46
- per_device_eval_batch_size=4,
47
  num_train_epochs=3,
48
  weight_decay=0.01,
49
  save_total_limit=2,
 
42
  output_dir="./results",
43
  evaluation_strategy="epoch",
44
  learning_rate=2e-5,
45
+ per_device_train_batch_size=2,
46
+ per_device_eval_batch_size=2,
47
  num_train_epochs=3,
48
  weight_decay=0.01,
49
  save_total_limit=2,