源码链接: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 计算一次。