使用tensorflow填充序列时出现``MemoryError''

Question

我正在尝试在AWS实例“ g2.2xlarge”上训练我的模型，但是在尝试向序列中添加填充时却出现“ MemoryError”。

content_array = keras.preprocessing.sequence.pad_sequences(content_array, maxlen=max_sequence_length,
                                                              padding='post')

出现此错误：

Traceback (most recent call last):
  File "trainer.py", line 185, in <module>
    train()
  File "trainer.py", line 52, in train
    padding='post')
  File "/usr/local/lib/python2.7/dist-packages/tensorflow/python/keras/preprocessing/sequence.py", line 94, in pad_sequences
    x = (np.ones((num_samples, maxlen) + sample_shape) * value).astype(dtype)
MemoryError

知道为什么吗？我什至还没有开始训练模型。

Answer 1

我错误地计算了最大序列长度，从而导致数量巨大。更正后，我没有任何问题。