您的会话因未知原因崩溃 - Google Colab

时间:2021-02-03 12:06:41

标签: python android tensorflow tensorflow-lite

当我尝试编译和训练模型时出现错误,例如:您的会话因未知原因崩溃并重新启动 google colab。

参考链接:https://medium.com/analytics-vidhya/tensorflow-lite-on-android-2f57267f7297

帮我解决。

  • 编译和训练模型
model.compile(optimizer=tf.keras.optimizers.Adam(),
loss='categorical_crossentropy',
metrics=['accuracy'])
ep = 25
steps_per_epoch = numpy.asarray(train_generator.n / batch_size)
validation_steps = numpy.asarray(validation_generator.n / batch_size)
history = model.fit(train_generator,
steps_per_epoch=steps_per_epoch,
epochs=ep,
validation_data=validation_generator,
validation_steps=validation_steps)```

0 个答案:

没有答案