我正在使用TensorflowLite model maker,我想知道如何在训练期间定期保存检查点,以便使用tensorboard检查损失图。如何做到这一点?
当我想训练时,我会调用这个函数,但似乎没有额外的设置,如写入检查点或为其指定目录。
第一个月
我正在使用TensorflowLite model maker,我想知道如何在训练期间定期保存检查点,以便使用tensorboard检查损失图。如何做到这一点?
当我想训练时,我会调用这个函数,但似乎没有额外的设置,如写入检查点或为其指定目录。
第一个月
2条答案
按热度按时间djp7away1#
好了,我发现了它是如何工作的!参数
grad_checkpoint=true
必须在指定给object_detector.create()函数的spec对象中设置!在spec中,模型检查点目录也可以指定!这就是我如何让它工作的:字符串
tktrz96b2#
TensorFlow Lite Model Maker不提供在训练期间保存检查点的直接方法,但您可以使用tensorboard。尝试以下操作:
字符串
如果你需要更多关于tensorboard的信息,请告诉我。