158 lines
3.9 KiB
YAML
158 lines
3.9 KiB
YAML
run_title: b18_ffc075_batch8x15
|
|
training_model:
|
|
kind: default
|
|
visualize_each_iters: 1000
|
|
concat_mask: true
|
|
store_discr_outputs_for_vis: true
|
|
losses:
|
|
l1:
|
|
weight_missing: 0
|
|
weight_known: 10
|
|
perceptual:
|
|
weight: 0
|
|
adversarial:
|
|
kind: r1
|
|
weight: 10
|
|
gp_coef: 0.001
|
|
mask_as_fake_target: true
|
|
allow_scale_mask: true
|
|
feature_matching:
|
|
weight: 100
|
|
resnet_pl:
|
|
weight: 30
|
|
weights_path: ${env:TORCH_HOME}
|
|
|
|
optimizers:
|
|
generator:
|
|
kind: adam
|
|
lr: 0.001
|
|
discriminator:
|
|
kind: adam
|
|
lr: 0.0001
|
|
visualizer:
|
|
key_order:
|
|
- image
|
|
- predicted_image
|
|
- discr_output_fake
|
|
- discr_output_real
|
|
- inpainted
|
|
rescale_keys:
|
|
- discr_output_fake
|
|
- discr_output_real
|
|
kind: directory
|
|
outdir: /group-volume/User-Driven-Content-Generation/r.suvorov/inpainting/experiments/r.suvorov_2021-04-30_14-41-12_train_simple_pix2pix2_gap_sdpl_novgg_large_b18_ffc075_batch8x15/samples
|
|
location:
|
|
data_root_dir: /group-volume/User-Driven-Content-Generation/datasets/inpainting_data_root_large
|
|
out_root_dir: /group-volume/User-Driven-Content-Generation/${env:USER}/inpainting/experiments
|
|
tb_dir: /group-volume/User-Driven-Content-Generation/${env:USER}/inpainting/tb_logs
|
|
data:
|
|
batch_size: 15
|
|
val_batch_size: 2
|
|
num_workers: 3
|
|
train:
|
|
indir: ${location.data_root_dir}/train
|
|
out_size: 256
|
|
mask_gen_kwargs:
|
|
irregular_proba: 1
|
|
irregular_kwargs:
|
|
max_angle: 4
|
|
max_len: 200
|
|
max_width: 100
|
|
max_times: 5
|
|
min_times: 1
|
|
box_proba: 1
|
|
box_kwargs:
|
|
margin: 10
|
|
bbox_min_size: 30
|
|
bbox_max_size: 150
|
|
max_times: 3
|
|
min_times: 1
|
|
segm_proba: 0
|
|
segm_kwargs:
|
|
confidence_threshold: 0.5
|
|
max_object_area: 0.5
|
|
min_mask_area: 0.07
|
|
downsample_levels: 6
|
|
num_variants_per_mask: 1
|
|
rigidness_mode: 1
|
|
max_foreground_coverage: 0.3
|
|
max_foreground_intersection: 0.7
|
|
max_mask_intersection: 0.1
|
|
max_hidden_area: 0.1
|
|
max_scale_change: 0.25
|
|
horizontal_flip: true
|
|
max_vertical_shift: 0.2
|
|
position_shuffle: true
|
|
transform_variant: distortions
|
|
dataloader_kwargs:
|
|
batch_size: ${data.batch_size}
|
|
shuffle: true
|
|
num_workers: ${data.num_workers}
|
|
val:
|
|
indir: ${location.data_root_dir}/val
|
|
img_suffix: .png
|
|
dataloader_kwargs:
|
|
batch_size: ${data.val_batch_size}
|
|
shuffle: false
|
|
num_workers: ${data.num_workers}
|
|
visual_test:
|
|
indir: ${location.data_root_dir}/korean_test
|
|
img_suffix: _input.png
|
|
pad_out_to_modulo: 32
|
|
dataloader_kwargs:
|
|
batch_size: 1
|
|
shuffle: false
|
|
num_workers: ${data.num_workers}
|
|
generator:
|
|
kind: ffc_resnet
|
|
input_nc: 4
|
|
output_nc: 3
|
|
ngf: 64
|
|
n_downsampling: 3
|
|
n_blocks: 18
|
|
add_out_act: sigmoid
|
|
init_conv_kwargs:
|
|
ratio_gin: 0
|
|
ratio_gout: 0
|
|
enable_lfu: false
|
|
downsample_conv_kwargs:
|
|
ratio_gin: ${generator.init_conv_kwargs.ratio_gout}
|
|
ratio_gout: ${generator.downsample_conv_kwargs.ratio_gin}
|
|
enable_lfu: false
|
|
resnet_conv_kwargs:
|
|
ratio_gin: 0.75
|
|
ratio_gout: ${generator.resnet_conv_kwargs.ratio_gin}
|
|
enable_lfu: false
|
|
discriminator:
|
|
kind: pix2pixhd_nlayer
|
|
input_nc: 3
|
|
ndf: 64
|
|
n_layers: 4
|
|
evaluator:
|
|
kind: default
|
|
inpainted_key: inpainted
|
|
integral_kind: ssim_fid100_f1
|
|
trainer:
|
|
kwargs:
|
|
gpus: -1
|
|
accelerator: ddp
|
|
max_epochs: 200
|
|
gradient_clip_val: 1
|
|
log_gpu_memory: None
|
|
limit_train_batches: 25000
|
|
val_check_interval: ${trainer.kwargs.limit_train_batches}
|
|
log_every_n_steps: 1000
|
|
precision: 32
|
|
terminate_on_nan: false
|
|
check_val_every_n_epoch: 1
|
|
num_sanity_val_steps: 8
|
|
limit_val_batches: 1000
|
|
replace_sampler_ddp: false
|
|
checkpoint_kwargs:
|
|
verbose: true
|
|
save_top_k: 5
|
|
save_last: true
|
|
period: 1
|
|
monitor: val_ssim_fid100_f1_total_mean
|
|
mode: max
|