Skip to content

GitLab

Commit 95cde5a9 authored by Geewook Kim's avatar Geewook Kim
Browse files

fix: update yaml, related to #29

parent 8997ed5c
Hide whitespace changes
Inline Side-by-side
Showing with 4 additions and 4 deletions
config/train_cord.yaml
View file @ 95cde5a9
...@@ -14,7 +14,7 @@ lr: 3e-5 ...@@ -14,7 +14,7 @@ lr: 3e-5
warmup_steps: 300 # 800/8*30/10, 10% warmup_steps: 300 # 800/8*30/10, 10%
num_training_samples_per_epoch: 800 num_training_samples_per_epoch: 800
max_epochs: 30 max_epochs: 30
max_steps: null max_steps: -1
num_workers: 8 num_workers: 8
val_check_interval: 1.0 val_check_interval: 1.0
check_val_every_n_epoch: 3 check_val_every_n_epoch: 3
......
config/train_docvqa.yaml
View file @ 95cde5a9
...@@ -15,7 +15,7 @@ lr: 3e-5 ...@@ -15,7 +15,7 @@ lr: 3e-5
warmup_steps: 10000 warmup_steps: 10000
num_training_samples_per_epoch: 39463 num_training_samples_per_epoch: 39463
max_epochs: 300 max_epochs: 300
max_steps: null max_steps: -1
num_workers: 8 num_workers: 8
val_check_interval: 1.0 val_check_interval: 1.0
check_val_every_n_epoch: 1 check_val_every_n_epoch: 1
......
config/train_rvlcdip.yaml
View file @ 95cde5a9
...@@ -15,7 +15,7 @@ lr: 2e-5 ...@@ -15,7 +15,7 @@ lr: 2e-5
warmup_steps: 10000 warmup_steps: 10000
num_training_samples_per_epoch: 320000 num_training_samples_per_epoch: 320000
max_epochs: 100 max_epochs: 100
max_steps: null max_steps: -1
num_workers: 8 num_workers: 8
val_check_interval: 1.0 val_check_interval: 1.0
check_val_every_n_epoch: 1 check_val_every_n_epoch: 1
......
config/train_zhtrainticket.yaml
View file @ 95cde5a9
...@@ -14,7 +14,7 @@ lr: 3e-5 ...@@ -14,7 +14,7 @@ lr: 3e-5
warmup_steps: 300 warmup_steps: 300
num_training_samples_per_epoch: 1368 num_training_samples_per_epoch: 1368
max_epochs: 10 max_epochs: 10
max_steps: null max_steps: -1
num_workers: 8 num_workers: 8
val_check_interval: 1.0 val_check_interval: 1.0
check_val_every_n_epoch: 1 check_val_every_n_epoch: 1
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment