Tensorflow 2在Colab中突然停止训练

时间:2019-11-10 16:26:43

标签: training-data tensorflow2.0

我刚刚注意到我的模型没有训练并在第1阶段停止?

history = model.fit(X_modified, ys, epochs = 500, batch_size=1600,verbose=1 )

这是Colab输出:

  

训练1616018个样本纪元1/500 1600/1616018   [.....................]-预计到达时间:3:42

,然后停止。我将运行时设置为带有GPU的Python 3。但是以下内容似乎矛盾吗?

sess = tf.compat.v1.Session(config=tf.compat.v1.ConfigProto(allow_soft_placement=True, 
                                        log_device_placement=True))
sess.close()

Device mapping:
/job:localhost/replica:0/task:0/device:XLA_CPU:0 -> device: XLA_CPU device

为什么不说GPU?

谢谢

0 个答案:

没有答案