configs/modules/noise_optimizer.yaml

noise_optimizer:
  target: mld.models.architectures.dno.DNO
  params:
    optimize: false
    max_train_steps: 400
    learning_rate: 0.1
    lr_scheduler: 'cosine'
    lr_warmup_steps: 50
    clip_grad: true
    loss_hint_type: 'l2'
    loss_diff_penalty: 0.000
    loss_correlate_penalty: 100
    visualize_samples: 0
    visualize_ske_steps: []
    output_dir: ${output_dir}