Tensorflow：当我拟合模型时，运行时断开

Question

我正在尝试在Google的colab中对文本进行分类。

model =tf.keras.Sequential([
    tf.keras.layers.Flatten(),
    tf.keras.layers.Dense(128, activation = 'relu'),
    tf.keras.layers.Dense(2, activation = 'sigmoid')
])
model.compile(optimizer='adam',loss='sparse_categorical_crossentropy',              metrics=['accuracy'])

但是，当我运行它时

model.fit(baseline_training, baseline_training_labels, epochs = 50)

我在colab的浏览器中断开了运行时连接！它重新连接了，但是当我在上面的单元格中运行时，再次给了我“运行时断开”的提示。

请问怎么了？

CS