error in line 778, networks.py

[wzmsltw]

Hi junleen

I tried to run the training code, but occur an error in

RainNet/models/networks.py

Line 776 in debaf7f

feat_l, feat_g = torch.cat([xf, xb])

File "/ssd3/vis/lintianwei/project/harmonization/RainNet-main/models/networks.py", line 778, in forward
feat_l, feat_g = torch.cat([xf, xb])
ValueError: too many values to unpack (expected 2)

Actually, feat_l and feat_g are not used during training. Is this a bug?

[junleen]

Sorry to make some unexpected changes. Please change the code to:

def forward(self, input, mask=None, gp=False, feat_loss=False):

        x, xf, xb = self.D(input, mask)
        feat_l, feat_g = torch.cat([xf, xb]), x