I doubt if the input data I trained is not a problem？

[695874419]

The forward method of InpaintingModel is as follows： def forward(self, images, edges, masks): images_masked = (images * (1 - masks).float()) + masks inputs = torch.cat((images_masked, edges), dim=1) outputs = self.generator(inputs) return outputs I visualized the variables about the image. As shown below： I think there is a problem with my images_masked variable. Is this input correct?

[anshen666]

你好，我最近也在跑这个代码。可以加你交流一下吗？我的微信：loveanshen 我的QQ：519838354 我的邮箱：[email protected] 非常期待你百忙中的回复