amihai85 commited on
Commit
aeebf0e
·
verified ·
1 Parent(s): 0c596e3

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -20,7 +20,7 @@ tokenized_dataset = dataset.map(tokenize_function, batched=True)
20
  training_args = TrainingArguments(
21
  output_dir="./results",
22
  overwrite_output_dir=True,
23
- evaluation_strategy="epoch",
24
  learning_rate=5e-5,
25
  per_device_train_batch_size=2,
26
  num_train_epochs=3,
 
20
  training_args = TrainingArguments(
21
  output_dir="./results",
22
  overwrite_output_dir=True,
23
+ eval_strategy="epoch", # Updated
24
  learning_rate=5e-5,
25
  per_device_train_batch_size=2,
26
  num_train_epochs=3,