YOLOv3 安装与环境配置

源码链接:https://github.com/eriklindernoren/PyTorch-YOLOv3

poetry 学习:https://notes.zhengxinonly.com/environment/use-poetry.html
主要是这段话,这两个文件的关系和作用,其中poetry.lock文件相当与requirement.txt文件.
在这里插入图片描述

poetry install

运行detect测试环境是否正确安装
$ poetry run yolo-detect --image data/samples/

报错 No module named ‘packaging.metadata’
原因:packaging包的版本太低,与poetry不适配。
解决: pip install packaging==24.1

报错 NVIDIA GeForce RTX 4090 with CUDA capability sm_89 is not compatible with the current PyTorch installation.
原因:poetry.lock(相当于requirements.txt文件)默认安装的torch版本过低,与4090的cuda驱动不适配。
解决:直接pip install个高版本的torch,pip install torch1.13.1+cu117 torchvision0.14.1+cu117 torchaudio==0.13.1 --extra-index-url https://download.pytorch.org/whl/cu117

TensorBoard train.loss横坐标

数值到10k。横坐标值是 epoch×batch数目,相当于在训练时他不是每个epoch算一个loss,而是对每一个batch算一个loss
最大到9932,batch_size是默认值,我也不知道是多少,算出来batch_num=33,每个epoch要load batch 33次,33*300=9900,算下来差不多的。

而 val 是每个 epoch 计算一次。

  • 1
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值