
debug: False
v3: False

# common 
device: 'cpu'       # 'cpu' / 'cuda:0' / 'cuda:1'  / 'cuda'

#dataset
train_dir: /mnt/ssd1T/hoanglv/Projects/KIE/DATA/dev_model/Receipt/train
val_dir: /mnt/ssd1T/hoanglv/Projects/KIE/DATA/dev_model/Receipt/test
slice_interval: 75
postprocess_type: receipt_postprocess

classes: [
    "Seller_company_name_value",
    "ID_value",
    "Date_value",
    "Total_key",
    "Total_value",
    "Other"
]
sampling: true            # sampling window - fix long document (larger than 512 token)

#model 
img_size: 224            # fixed
max_seq_length: 512        # fixed
max_num_words: 150
tokenizer_weights: "/mnt/ssd1T/hoanglv/Projects/KIE/sdsvkie/microsoft/microsoft/layoutxlm-base"      # fixed
weights: "/mnt/ssd1T/hoanglv/Projects/KIE/sdsvkie/microsoft/microsoft/layoutxlm-base"

# opt + scheduler
batch_size: 8
epochs: 100
lr: 9.0e-6
scheduler: "linear"
shuffle: True
num_workers: 4
save_dir: workdirs/invoice/exp1
save_weight_interval: 10
eval_delay: 50
wandb: null 

# inference
inference_weights: null 
text_det: yolox-s-general-text-pretrain-20221226
text_reg: satrn-lite-general-pretrain-20230106

