Spaces:
Running
Running
run_title: '' | |
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} | |
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 | |
defaults: | |
- location: docker | |
- data: abl-04-256-mh-dist | |
- discriminator: pix2pixhd_nlayer | |
- optimizers: default_optimizers | |
- visualizer: directory | |
- evaluator: default_inpainted | |
- trainer: any_gpu_large_ssim_ddp_final | |
- hydra: overrides | |