Update README.md
Browse files
README.md
CHANGED
@@ -13,4 +13,28 @@ base_model:
|
|
13 |
|
14 |
| Model Version | F1 Score | EM Score | EM (no space) |
|
15 |
|--------------|----------|-----------|---------------|
|
16 |
-
| Initial LayoutLMv3 | 84.965 | 50.4323 | 50.4323 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
13 |
|
14 |
| Model Version | F1 Score | EM Score | EM (no space) |
|
15 |
|--------------|----------|-----------|---------------|
|
16 |
+
| Initial LayoutLMv3 | 84.965 | 50.4323 | 50.4323 |
|
17 |
+
|
18 |
+
```
|
19 |
+
training_args = TrainingArguments(
|
20 |
+
output_dir="test",
|
21 |
+
max_steps=1500,
|
22 |
+
per_device_train_batch_size=2,
|
23 |
+
per_device_eval_batch_size=2,
|
24 |
+
learning_rate=2e-5,
|
25 |
+
evaluation_strategy="steps",
|
26 |
+
eval_steps=100,
|
27 |
+
load_best_model_at_end=True,
|
28 |
+
metric_for_best_model="f1",
|
29 |
+
gradient_accumulation_steps=4 # 메모리 부족시 활용
|
30 |
+
)
|
31 |
+
|
32 |
+
trainer = Trainer(
|
33 |
+
model=model,
|
34 |
+
args=training_args,
|
35 |
+
train_dataset=train_dataset,
|
36 |
+
eval_dataset=eval_dataset,
|
37 |
+
tokenizer=processor,
|
38 |
+
data_collator=default_data_collator,
|
39 |
+
compute_metrics=compute_metrics,
|
40 |
+
)
|