Spaces:
Running
Running
| world_size: 4 | |
| experiment_name: 'FFHQH_SEHARM_1024_FINAL' | |
| datasets: | |
| iharmony: "/home/jovyan/datasets/iHarmony4/" | |
| crop_size: 1024 | |
| # dataset_root: "/home/jovyan/datasets/iHarmony4/" | |
| ffhqh: "/home/jovyan/datasets/synthetic_ffhq/" | |
| dataset_root: "/home/jovyan/datasets/synthetic_ffhq/" | |
| ffhq: "/home/jovyan/datasets/ffhq/images1024x1024/" | |
| isTrain: True | |
| task_for_oneformer: "matting" | |
| datasets_to_use_test: ["FFHQH"] | |
| #"IhdDataset", "HCOCO", "HFlickr", "Hday2night", "HAdobe5k", "FFHQH"] | |
| dataset_to_use: "FFHQH" | |
| pretrained_model: 'microsoft/swin-base-patch4-window7-224' #"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" | |
| load_pretrained: True | |
| # checkpoint: "/home/jovyan/efremyan/harmonization/best_50.pth" | |
| # checkpoint: "/home/jovyan/efremyan/harmonization/checkpoints/01_August_13_38/epoch-13.pth" | |
| checkpoint: "/home/jovyan/efremyan/harmonization/checkpoints/03_August_23_36/epoch-42.pth" | |
| distributed_addr: "localhost" | |
| # distributed_port: "12356" | |
| # image_size: 512 | |
| lr: 1e-5 | |
| epochs: 300 | |
| disable_validation: False | |
| disable_mixed_precision: False | |
| log_image_interval: 10 | |
| log_image_number: 4 | |
| save_model_interval: 1 | |
| lambda_losses: | |
| default: 1. | |
| Color: 0. | |
| PSNR: 1. | |
| LPIPS: 0.5 | |
| L2: 1. | |
| L1: 0. | |
| softmax: 1e-4 | |
| model: | |
| grid_counts: [10, 5, 1] | |
| enc_sizes: [3, 16, 32, 64, 128, 256, 512] | |
| init_weights: [0.5, 0.5] # weights of local grided image and global harmonized | |
| init_value: 0.8 | |
| skips: True | |