23 lines
625 B
YAML
23 lines
625 B
YAML
|
dataset_path: "Zombely/fiszki-ocr-train"
|
||
|
|
pretrained_model_path: "Zombely/plwiki-proto-fine-tuned-v2"
|
||
|
|
start_model_path: "Zombely/plwiki-proto-fine-tuned-v2"
|
||
|
|
output_model_path: "Zombely/plwiki-proto-fine-tuned-v3"
|
||
|
|
wandb_test_name: "fiszki-ocr-fine-tune"
|
||
|
|
checkpoint_path: "./checkpoint"
|
||
|
|
max_length: 768
|
||
|
|
image_size: [1920, 2560]
|
||
|
|
train_config:
|
||
|
|
max_epochs: 1
|
||
|
|
val_check_interval: 0.5
|
||
|
|
check_val_every_n_epoch: 1
|
||
|
|
gradient_clip_val: 1.0
|
||
|
|
num_training_samples_per_epoch: 800
|
||
|
|
lr: 3.0e-5
|
||
|
|
train_batch_sizes: [8]
|
||
|
|
val_batch_sizes: [1]
|
||
|
|
seed: 2022
|
||
|
|
num_nodes: 1
|
||
|
|
warmup_steps: 300
|
||
|
|
result_path: "./result"
|
||
|
|
verbose: True
|