Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -45,7 +45,7 @@ if st.button("トレーニング開始"):
|
|
| 45 |
# トレーニングè¨å®š
|
| 46 |
training_args = TrainingArguments(
|
| 47 |
output_dir=output_dir,
|
| 48 |
-
|
| 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,
|