提交 a7f4fcef 编写于 作者: W wuxing03

fix error occured by update api to 1.8

上级 efb32c74
...@@ -233,7 +233,7 @@ class VAE(object): ...@@ -233,7 +233,7 @@ class VAE(object):
layers.sampling_id( layers.sampling_id(
layers.softmax( layers.softmax(
layers.squeeze(output_layer(x),[1]) layers.squeeze(output_layer(x),[1])
),dtype='int'), ),dtype='int'),
depth=self.tar_vocab_size), [1]) depth=self.tar_vocab_size), [1])
if mode == 'train': if mode == 'train':
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册