File size: 797 Bytes
0f90f73 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 |
indir: no # to be overriden in CLI
outdir: no # to be overriden in CLI
model:
path: no # to be overriden in CLI
checkpoint: model_zoo/big_lama.ckpt
dataset:
kind: default
img_suffix: .png
pad_out_to_modulo: 8
device: cuda
out_key: inpainted
refine: False # refiner will only run if this is True
refiner:
gpu_ids: 0,1 # the GPU ids of the machine to use. If only single GPU, use: "0,"
modulo: ${dataset.pad_out_to_modulo}
n_iters: 15 # number of iterations of refinement for each scale
lr: 0.002 # learning rate
min_side: 512 # all sides of image on all scales should be >= min_side / sqrt(2)
max_scales: 3 # max number of downscaling scales for the image-mask pyramid
px_budget: 1800000 # pixels budget. Any image will be resized to satisfy height*width <= px_budget |