mirror of
https://github.com/YaoFANGUK/video-subtitle-remover.git
synced 2026-05-14 07:37:31 +08:00
37 lines
753 B
YAML
37 lines
753 B
YAML
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}
|
|
|
|
defaults:
|
|
- location: docker
|
|
- data: abl-04-256-mh-dist
|
|
- generator: pix2pixhd_multidilated_catin_4dil_9b
|
|
- discriminator: pix2pixhd_nlayer
|
|
- optimizers: default_optimizers
|
|
- visualizer: directory
|
|
- evaluator: default_inpainted
|
|
- trainer: any_gpu_large_ssim_ddp_final_benchmark
|
|
- hydra: overrides
|