生成的图片不太对
SeeU1119 opened this issue · comments
很麻烦再一次打扰您, 我按照readme的教程从头开始训练,迭代了一段时间,发现生成的fake_A总是将脸部变成黑色,请问一下,我是哪里出了什么问题,谢谢。
就比如下面所示:
还有我发现在得到weight使用稀疏张量会减慢模型训练速度,是否可以注释掉:
weight = weight[weight_ind[0], weight_ind[1], weight_ind[2]]
ret = torch.sparse.FloatTensor(weight_ind, weight, torch.Size([3, HW, HW]))
y = torch.bmm(weight.to_dense(), g_source)--------> y = torch.bmm(weight, g_source)
我也是。 只有0_0_fake,jpg 不是黑的 别的第三张 Generator的图面部全部都是黑色的
请使用pytorch 1.5
谢谢您的回复, 我先去试试!