Dongwookss commited on
Commit
2ea1f41
·
verified ·
1 Parent(s): c948e78

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +25 -1
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
+ )