mirror of
https://github.com/allenai/olmocr.git
synced 2025-11-15 18:08:41 +00:00
Loss config
This commit is contained in:
parent
a3997419b3
commit
f8071c7457
@ -84,7 +84,7 @@ training:
|
|||||||
save_steps: 500
|
save_steps: 500
|
||||||
save_total_limit: 3
|
save_total_limit: 3
|
||||||
load_best_model_at_end: true
|
load_best_model_at_end: true
|
||||||
metric_for_best_model: eval_loss
|
metric_for_best_model: eval_processed_00_documents_eval_s2pdf_loss
|
||||||
greater_is_better: false
|
greater_is_better: false
|
||||||
|
|
||||||
report_to:
|
report_to:
|
||||||
|
|||||||
Loading…
x
Reference in New Issue
Block a user