Pramod
@derrick-xwp did you happen to resolve this issue? I am curious since I encounter the same bug.
@uuzgu and @nyj-ocean do something like this: ``` 179 scores = pack_padded_sequence(scores, decode_lengths, batch_first=True).data 180 targets = pack_padded_sequence(targets, decode_lengths, batch_first=True).data ```
One quick hack I found was to use the [result](https://github.com/switchablenorms/CelebAMask-HQ/blob/3f6c45df5e67130568f9459f24f8d6a5ff836d30/MaskGAN_demo/demo.py#L203) and pass it the variable `self.output_img`
Yes, I have the same doubt, it would be really helpful to know. Thank you