【发布时间】:2017-06-21 20:43:02
【问题描述】:
我正在尝试使用 LSTMs 实现 基本翻译模型,其中输入和输出是 CNTK 中不同语言的句子。
为了实现这一点,我正在创建如下模型:
def create_model(x):
with c.layers.default_options():
m = c.layers.Recurrence(c.layers.LSTM(input_vocab_size))(x)
m = sequence.last(m)
y = c.layers.Recurrence(c.layers.LSTM(label_vocab_size))(m)
return m
batch_axis = Axis.default_batch_axis()
input_seq_axis = Axis('inputAxis')
input_dynamic_axes = [batch_axis, input_seq_axis]
raw_input = input_variable(shape = (input_vocab_dim), dynamic_axes = input_dynamic_axes, name = 'raw_input')
z= create_model(raw_input)
但我收到以下错误:
RuntimeError: Currently PastValue/FutureValue Function only supports input operand with 2 dynamic axis (1 sequence-axis and 1 batch-axis)
据我了解,动态轴基本上是在数据加载后决定的那些轴,在这种情况下,批量大小和输入句子的长度。我认为我不会在任何地方更改输入的动态轴。
非常感谢任何帮助。
【问题讨论】:
标签: deep-learning lstm cntk