tensorflow 如何在tflite_model_maker中使用“tensorboard”?可以在tflite_model_maker中使用custom_callbacks吗?

62o28rlo  于 2023-06-24  发布在  其他
关注(0)|答案(1)|浏览(100)

我已经使用了“tf.keras.callbacks.tensorboard”和我的自定义keras-models。但是我不能将它与tflite_model_maker一起用于对象检测。如果可能的话,请让我知道……谢谢!
我找不到关于documentation的任何信息。

kgsdhlau

kgsdhlau1#

Tensorboard目录由EfficientDetModelSpec类配置。需要参数model_dir,会有保存的tensorboard日志沿着模型检查点

spec = model_spec.get('efficientdet_lite0', model_dir='mytbdir')
model = object_detector.create(train_data, model_spec=spec,...)

!ls mytbdir
checkpoint ckpt-1.data-00000-of-00001 ckpt-1.index
config.yaml train train_images

// usage from colab:
%load_ext tensorboard
%tensorboard --logdir tbdir

相关问题