yolo - 如何恢复 Yolov3 训练？

Question

我是深度学习的新手，我有一个 yolov3 模型，我一直在用我的自定义数据进行训练。每次训练，训练似乎都是从零开始。如何让模型从上次停止的地方继续训练？我的设置与此 repo相同。

标签： yoloresuming-training

Answer 1

您可以model.load_weights(path_to_checkpoint)在模型定义后立即使用line 41intrain.py并在您离开的地方继续训练