search

TJU-Zhaoxy / A-Pyramid-Architecture-of-GANs

Learning Face Age Progression: A Pyramid Architecture of GANs
28 stars 10 forks source link

generate的输出 #3

Open jilner opened 5 years ago

jilner commented 5 years ago

生成器的输出好像不是224，计算出来是bs 3 221*221?

TJU-Zhaoxy commented 5 years ago

对的，确实是这样。所以在计算pixel loss的时候做了限制，现在还有一些不合理的地方，在之后有时间会进行改进。

jilner commented 5 years ago

对的，确实是这样。所以在计算pixel loss的时候做了限制，现在还有一些不合理的地方，在之后有时间会进行改进。

还有模型保存的时候 我用的0.4.1 这样torch.save(self.D.parameters()是会报错的，再冒昧问一下，大神用的什么卡啊，我的1080 跑不了就outmemory了

TJU-Zhaoxy commented 5 years ago

对，我现在也是存在这个问题，所以想等更好的服务器来了再进行测试。我感觉可能需要10G往上显存了。也可能是代码哪里有问题，我也是个初学者，相互交流吧