当我尝试编译和训练模型时出现错误,例如:您的会话因未知原因崩溃并重新启动 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)```