diff --git a/recipes/srflow/train_div2k_srflow.yml b/recipes/srflow/train_div2k_srflow.yml index 4b7522ae..05d55cad 100644 --- a/recipes/srflow/train_div2k_srflow.yml +++ b/recipes/srflow/train_div2k_srflow.yml @@ -35,14 +35,12 @@ networks: initial_stride: 2 flow_scale: 4 train_RRDB: false # <-- Start false. After some time, ~20k-50k steps, set to true. TODO: automate this. - train_RRDB_delay: 0.5 pretrain_rrdb: ../experiments/pretrained_rrdb.pth # <-- Insert path to your pretrained RRDB here. flow: patch_size: 160 K: 16 L: 3 - act_norm_start_step: 100 noInitialInj: true coupling: CondAffineSeparatedAndCond additionalFlowNoAffine: 2