【问题标题】:How to Resume Yolov3 training?如何恢复 Yolov3 训练?
【发布时间】:2021-06-24 23:37:35
【问题描述】:

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

【问题讨论】:

    标签: yolo resuming-training


    【解决方案1】:

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

    【讨论】:

    • 谢谢!我假设 'path_to_checkpoint' 将是 './"yolov3"' ?因为训练会生成一个 'yolov3.data-00000-of-00001' 文件,我无法将该名称输入到 load_weights 方法中。我在脚本中看到了 model.save_weights(./"yolov3")。
    猜你喜欢
    • 2021-12-20
    • 1970-01-01
    • 2019-07-15
    • 2020-12-29
    • 2021-05-31
    • 1970-01-01
    • 2022-07-09
    • 2016-02-18
    相关资源
    最近更新 更多