【问题标题】:Google Colab Notebook completely freezes when training a YOLO model训练 YOLO 模型时,Google Colab Notebook 完全冻结
【发布时间】:2021-02-12 08:59:59
【问题描述】:

我正在学习使用 YOLO 训练自定义对象检测模型的教程。 This is the tutorial and also where I got the Notebook

在最后一个单元格的训练位之前一切正常。它完成了几次迭代,然后冻结并且页面停止响应。我有点着急,所以任何解决这个问题的方法都值得赞赏。 TIA

【问题讨论】:

  • 在collab中,你是把runtime设置为GPU还是TPU?
  • @jakub 使用 GPU
  • 你能发一张它在哪里冻结的图片吗?

标签: python machine-learning google-colaboratory object-detection yolo


【解决方案1】:
from IPython.utils import io
with io.capture_output() as captured:
   !./darknet detector train data/obj.data cfg/yolov3_training.cfg darknet53.conv.74 - 
   dont_show

我有同样的问题,这个对我有用!

【讨论】:

    【解决方案2】:

    问题在于,Darknet 提供了太多的输出,以至于网页达到了内存限制。我通过每隔几分钟清除一次输出解决了这个问题。

    【讨论】:

    • 你能举一个例子来说明如何做到这一点吗?你的答案和你的问题不够详细。还是谢谢。
    猜你喜欢
    • 2021-02-21
    • 1970-01-01
    • 1970-01-01
    • 1970-01-01
    • 1970-01-01
    • 2020-06-05
    • 2020-03-26
    • 1970-01-01
    • 2022-01-08
    相关资源
    最近更新 更多