Sakalti commited on
Commit
4a4175e
•
1 Parent(s): cfd58cc

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -45,7 +45,7 @@ if st.button("トレーニング開始"):
45
  # トレーニング設定
46
  training_args = TrainingArguments(
47
  output_dir=output_dir,
48
- evaluation_strategy="epoch",
49
  learning_rate=2e-5,
50
  per_device_train_batch_size=8,
51
  num_train_epochs=1,
 
45
  # トレーニング設定
46
  training_args = TrainingArguments(
47
  output_dir=output_dir,
48
+ eval_strategy="epoch",
49
  learning_rate=2e-5,
50
  per_device_train_batch_size=8,
51
  num_train_epochs=1,