使用pix2pix网络进行stackgan的叠加
pix2pix来自github上一位日本同学的代码
stackgan:https://github.com/hanzhanggit/StackGAN-Pytorch/blob/master/code/main.py
先留一个我的问题,等过几天测试来回答,感觉应该是我的一个锅,从论文里的阅读不够仔细,关于这两个网络的训练顺序,我现在的代码训练是每个epoch,串行训练两个网络,但是经过学长提醒以及上述代码的参考,应该是训练完第一层网络再开始下一层,而不是每次两层一起训练。
即:原始:
for epoch = 1:max_epoch
训练第一层;
输入第一层,输出结果给第二层
训练第二层
end
目前准确率83
改为:
for epoch = 1:max_epoch
训练第一层;
end
for epoch = 1:max_epoch
读取第一层训练的最后一次模型
输入给上述模型,输出结果
拿输出结果给,训练第二层;
end
未完待续