Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
@@ -111,7 +111,8 @@ if (runModel=='1'):
|
|
111 |
weight_decay=0.02,
|
112 |
logging_dir='./logs_' + modelNameToUse,
|
113 |
logging_steps=10,
|
114 |
-
evaluation_strategy="epoch",
|
|
|
115 |
load_best_model_at_end=True, # Load the best model based on evaluation
|
116 |
)
|
117 |
|
|
|
111 |
weight_decay=0.02,
|
112 |
logging_dir='./logs_' + modelNameToUse,
|
113 |
logging_steps=10,
|
114 |
+
evaluation_strategy="epoch", # Evaluation strategy is 'epoch'
|
115 |
+
save_strategy="epoch", # Save strategy should also be 'epoch'
|
116 |
load_best_model_at_end=True, # Load the best model based on evaluation
|
117 |
)
|
118 |
|