Update srflow recipe

This commit is contained in:
James Betker 2020-12-16 10:36:27 -07:00
parent 49327b99fe
commit daee1b5572

View File

@ -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