加载预训练砝码后立即终止

时间:2019-11-28 11:21:31

标签: object-detection yolo

我正在使用Yolov3解决检测问题。我试图在自己的数据集上训练模型。不幸的是,在加载了预训练的权重之后,该过程终止并返回到Ubuntu终端。我附在终端输出下方。

Loading weights from darknet53.conv.74...darknet-yolov3
Done!
user@tf-deeplearning-1:~/Downloads/darknet$

我试图解决的问题是:

  • 调整“ yolo”配置文件中的高度和宽度,
  • 在darknet的Makefile中更改GPU架构配置(我正在使用Tesla P4),
  • 制作带有和不带有CUDNN的文件,这意味着CUDNN = 0和CUDNN = 1,但是没有运气。

有人曾经遭受过这个问题吗?请在这件事上给予我帮助!高度赞赏。

0 个答案:

没有答案