【发布时间】:2021-06-24 23:37:35
【问题描述】:
我是深度学习的新手,我有一个 yolov3 模型,我一直在用我的自定义数据进行训练。每次训练,训练似乎都是从零开始。如何让模型从上次停止的地方继续训练? 我的设置和this repo一样。
【问题讨论】:
我是深度学习的新手,我有一个 yolov3 模型,我一直在用我的自定义数据进行训练。每次训练,训练似乎都是从零开始。如何让模型从上次停止的地方继续训练? 我的设置和this repo一样。
【问题讨论】:
您可以在train.py 中的line 41 定义模型后使用model.load_weights(path_to_checkpoint),并在您离开的地方继续训练
【讨论】: