未验证 提交 90acc7c1 编写于 作者: L lvmengsi 提交者: GitHub

fix readme (#2677)

* Update README.md
上级 7d614817
......@@ -18,7 +18,9 @@
本图像生成模型库包含CGAN\[[3](#参考文献)\], DCGAN\[[4](#参考文献)\], Pix2Pix\[[5](#参考文献)\], CycleGAN\[[6](#参考文献)\], StarGAN\[[7](#参考文献)\], AttGAN\[[8](#参考文献)\], STGAN\[[9](#参考文献)\]
注意:AttGAN和STGAN的网络结构中,判别器去掉了instance norm。
注意:
1. AttGAN和STGAN的网络结构中,判别器去掉了instance norm。
2. StarGAN,AttGAN和STGAN由于梯度惩罚所需的操作目前只支持GPU,需使用GPU训练。
图像生成模型库库的目录结构如下:
```
......
......@@ -308,12 +308,8 @@ class StarGAN(object):
loss_name=dis_trainer.d_loss.name,
build_strategy=build_strategy)
#losses = [[], []]
t_time = 0
test_program = gen_trainer.infer_program
utility.save_test_image(0, self.cfg, exe, place, test_program,
gen_trainer, self.test_reader)
for epoch_id in range(self.cfg.epoch):
batch_id = 0
for i in range(self.batch_num):
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册