未验证 提交 cc6d8757 编写于 作者: C ceci3 提交者: GitHub

Fix mistake (#4644)

* fix init

* update

* fix mistake
上级 d831712a
...@@ -370,7 +370,7 @@ def linear(input, ...@@ -370,7 +370,7 @@ def linear(input,
def conv_cond_concat(x, y): def conv_cond_concat(x, y):
batch = fluid.layers.shape(x)[0] batch = fluid.layers.shape(x)[0]
ones = fluid.layers.fill_constant( ones = fluid.layers.fill_constant(
shape=[ones, y.shape[1], x.shape[2], x.shape[3]], shape=[batch, y.shape[1], x.shape[2], x.shape[3]],
dtype="float32", dtype="float32",
value=1.0) value=1.0)
out = fluid.layers.concat([x, ones * y], 1) out = fluid.layers.concat([x, ones * y], 1)
......
python infer.py --model_net CGAN --init_model ./output/c_gan/checkpoints/19/ --n_samples 32 --noise_size 100 --output ./infer_result/c_gan/ python infer.py --model_net CGAN --init_model ./output/cgan/checkpoints/19/ --n_samples 32 --noise_size 100 --output ./infer_result/cgan/
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册