Fix bugs with srflow after refactor

codes/models/srflow/FlowStep.py

@@ -2,6 +2,8 @@ import torch
 from torch import nn as nn
 
 import models.srflow.Permutations
+import models.srflow.FlowAffineCouplingsAblation
+import models.srflow.FlowActNorms
 
 
 def getConditional(rrdbResults, position):

recipes/srflow/train_div2k_srflow.yml

@@ -24,12 +24,12 @@ datasets:
 networks:
   generator:
     type: generator
-    which_model_G: srflow_orig
+    which_model_G: srflow
     nf: 64
     nb: 23
     K: 16
     scale: 4
-    initial_stride: 2
+    initial_stride: 1
     flow_scale: 4
     train_RRDB: false    # <-- Start false. After some time, ~20k-50k steps, set to true. TODO: automate this.
     pretrain_rrdb: ../experiments/pretrained_rrdb.pth  # <-- Insert path to your pretrained RRDB here.