6b7b82ff60b8ca7a26b50e50210a06a46bdc848a,lib/prosr/models/generators.py,ProSR,forward,#ProSR#,131

Before Change

    except AttributeError:
      max_scale_idx = len(self.opt.scale) - 1

    if scale is None:
      scale = self.opt.scale[max_scale_idx]

    feats = self.get_init_conv(scale)(x)
    output = []
    for s in range(1, int(log2(scale))+1):
      if self.opt.residual_denseblock:

After Change

  def forward(self,x,upscale_factor=None, base_img=None, blend=1.0):

    if upscale_factor is None:
      upscale_factor = self.max_scale
    else:
      valid_upscale_factors = [2**(i+1) for i in range(self.n_denseblocks)]
      if not (1 < upscale_factor <= self.max_scale and upscale_factor % 2 == 0):
        error("Invalid upscaling factor: choose one of: {}".format(
          valid_upscale_factors))
        raise SystemExit(1)

    feats = self.get_init_conv(log2(upscale_factor))(x)
    output = []
    for s in range(1, int(log2(upscale_factor))+1):
      if self.residual_denseblock:

Instances