You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
def forward(self, x, get_steps=False):
fea = self.conv_first(x)
block_idxs = opt_get(self.opt, ['network_G', 'flow', 'stackRRDB', 'blocks']) or []
block_results = {}
for idx, m in enumerate(self.RRDB_trunk.children()):
fea = m(fea)
for b in block_idxs:
if b == idx:
block_results["block_{}".format(idx)] = fea
trunk = self.trunk_conv(fea)
last_lr_fea = fea + trunk
It seems that fea in the last_lr_fea=fea+trunk line is the feature of the last RRDB block before self.turnk_conv. But to my knowledge, the skip connection should be last_lr_fea = x+ trunk. Is my understanding right?
The text was updated successfully, but these errors were encountered:
In the forward function of the RRDBNet
It seems that
fea
in thelast_lr_fea=fea+trunk
line is the feature of the last RRDB block beforeself.turnk_conv
. But to my knowledge, the skip connection should belast_lr_fea = x+ trunk
. Is my understanding right?The text was updated successfully, but these errors were encountered: