mrq/DL-Art-School
1
3
Fork 5
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Update srflow recipe

Browse Source
This commit is contained in:
James Betker 2020-12-16 10:36:27 -07:00
parent 49327b99fe
commit daee1b5572
1 changed files with 0 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
recipes/srflow/train_div2k_srflow.yml
View File

@ -35,14 +35,12 @@ networks:
initial_stride: 2 initial_stride: 2
flow_scale: 4 flow_scale: 4
train_RRDB: false # <-- Start false. After some time, ~20k-50k steps, set to true. TODO: automate this. 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. pretrain_rrdb: ../experiments/pretrained_rrdb.pth # <-- Insert path to your pretrained RRDB here.
flow: flow:
patch_size: 160 patch_size: 160
K: 16 K: 16
L: 3 L: 3
act_norm_start_step: 100
noInitialInj: true noInitialInj: true
coupling: CondAffineSeparatedAndCond coupling: CondAffineSeparatedAndCond
additionalFlowNoAffine: 2 additionalFlowNoAffine: 2