我在Google Cloud ML上使用以下示例代码尝试训练我的模型:
import kerasfrom keras import optimizersfrom keras import lossesfrom keras import metricsfrom keras.models import Model, Sequentialfrom keras.layers import Dense, Lambda, RepeatVector, TimeDistributedimport numpy as npdef test(): model = Sequential() model.add(Dense(2, input_shape=(3,))) model.add(RepeatVector(3)) model.add(TimeDistributed(Dense(3))) model.compile(loss=losses.MSE, optimizer=optimizers.RMSprop(lr=0.0001), metrics=[metrics.categorical_accuracy], sample_weight_mode='temporal') x = np.random.random((1, 3)) y = np.random.random((1, 3, 3)) model.train_on_batch(x, y)if __name__ == '__main__': test()
我得到了以下错误:
副本主节点0以非零状态245退出。终止原因: 错误。
详细的错误输出内容很多,所以我把它粘贴到了这里的pastebin
回答:
问题已解决。我所做的只是使用tensorflow 1.1.0而不是默认的1.0.1