world_size: 1 
experiment_name: "test"
datasets:
    synthetic_fg: "/home/jovyan/datasets/synthetic_psi/"
    synthetic_animals: "/home/jovyan/datasets/synthetic_psiny/"
    bg: "/home/jovyan/datasets/matting/background/testval/"
    ppm100: "/home/jovyan/kvanchiani/stylegan3/PPM-100/image"
    aim500: "/home/jovyan/datasets/AIM-500"
    am2k: 
        train_original: "/home/jovyan/datasets/matting/am-2k/train/original"
        train_mask: "/home/jovyan/datasets/matting/am-2k/train/mask"
        background: "/home/jovyan/datasets/matting/am-2k/background/train"
        validation_original: "/home/jovyan/datasets/matting/am-2k/validation/original/"
        validation_mask: "/home/jovyan/datasets/matting/am-2k/validation/mask/"
        validation_trimap: "/home/jovyan/datasets/matting/am-2k/validation/trimap/"
    tiktok: "/home/jovyan/datasets/tiktokdataset/dataset"
    p3m10k: "/home/jovyan/datasets/matting/P3M-10k"
    p3m10k_test:
        VAL500P:
            ROOT_PATH: "P3M-500-P/"
            ORIGINAL_PATH: "P3M-500-P/blurred_image/"
            MASK_PATH: "P3M-500-P/mask/"
            TRIMAP_PATH: "P3M-500-P/trimap/"
            SAMPLE_NUMBER: 500
        VAL500NP:
            ROOT_PATH: "P3M-500-NP/"
            ORIGINAL_PATH: "P3M-500-NP/original_image/"
            MASK_PATH: "P3M-500-NP/mask/"
            TRIMAP_PATH: "P3M-500-NP/trimap/"
            SAMPLE_NUMBER: 500
    MAX_SIZE_H: 1600
    MAX_SIZE_W: 1600
    image_crop: 800
    max_image_count: 10000
dataset_to_use: MixedDataset
pretrained_model: "microsoft/swinv2-tiny-patch4-window8-256"  #"nielsr/mask2former-swin-base-youtubevis-2021" #"nvidia/mit-b2"
batch_size: 4 
num_workers: 4
log_dir: "log"
checkpoint_dir: "checkpoints"
checkpoint: "best-89.pth"
distributed_addr: "localhost"
distributed_port: "12357"
image_size: 800
lr: 1e-7
epochs: 200
disable_validation: False
warmup_steps: 2
validate_each_epoch: 5
max_images_for_validation: 500
disable_mixed_precision: True
log_image_interval: 500
log_image_number: 8
save_model_interval: 10000
switch: 3
lambda_losses: 
    default: 1.
    Laplassian: 1.
    Grad: 3.
    L1: 1.
    switch: 1e-6