yolov7关闭混合精度训练
时间: 2023-09-02 14:13:43 浏览: 522
yolov7代码及全部预训练权重
5星 · 资源好评率100%
要关闭YOLOv7的混合精度训练,你可以在训练命令中添加`--no-amp`参数。这将禁用混合精度训练,使模型以全精度进行训练。例如,使用以下命令进行训练:
```python
python train.py --workers 8 --device 0 --batch-size 32 --data data/dataset/data.yaml --img 640 640 --cfg cfg/training/yolov7.yaml --weights 'yolov7_training.pt' --name yolov7-custom --hyp data/hyp.scratch.custom.yaml --no-amp
```
这样,YOLOv7将以全精度进行训练,而不使用混合精度。<span class="em">1</span><span class="em">2</span><span class="em">3</span>
#### 引用[.reference_title]
- *1* *2* *3* [YOLO | 用YOLOv7训练自己的数据集(超详细版)](https://blog.csdn.net/weixin_44649780/article/details/128657943)[target="_blank" data-report-click={"spm":"1018.2226.3001.9630","extra":{"utm_source":"vip_chatgpt_common_search_pc_result","utm_medium":"distribute.pc_search_result.none-task-cask-2~all~insert_cask~default-1-null.142^v92^chatsearchT3_1"}}] [.reference_item style="max-width: 100%"]
[ .reference_list ]
阅读全文