微信公众号搜"智元新知"关注
微信扫一扫可直接关注哦!

训练YOLO模型时,Google Colab Notebook完全冻结

如何解决训练YOLO模型时,Google Colab Notebook完全冻结

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

一切正常,直到训练位(最后一个单元格)为止。它完成了几次迭代,然后冻结,页面停止响应。我有点着急,请以任何方式解决此问题。 TIA

解决方法

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

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

,

问题是Darknet提供了太多的输出,导致网页达到了内存限制。我每隔几分钟清除一次输出即可解决问题。

版权声明:本文内容由互联网用户自发贡献,该文观点与技术仅代表作者本人。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容, 请发送邮件至 dio@foxmail.com 举报,一经查实,本站将立刻删除。