Views
No views yet
| Model Version | F1 Score | EM Score | EM (no space) |
|---|---|---|---|
| Initial LayoutLMv3 | 84.965 | 50.4323 | 50.4323 |
training_args = TrainingArguments(
output_dir="test",
max_steps=1500,
per_device_train_batch_size=2,
per_device_eval_batch_size=2,
learning_rate=2e-5,
evaluation_strategy="steps",
eval_steps=100,
load_best_model_at_end=True,
metric_for_best_model="f1",
gradient_accumulation_steps=4 # 메모리 부족시 활용
)
trainer = Trainer(
model=model,
args=training_args,
train_dataset=train_dataset,
eval_dataset=eval_dataset,
tokenizer=processor,
data_collator=default_data_collator,
compute_metrics=compute_metrics,
)