YOLOv5集成了Weights & Biases Logging 🌟 NEW
Weights & Biases (W&B)是一个用于实时训练过程可视化和训练数据云端记录的工具,方便做训练数据的比较和研究,也方便团队协作。
详细使用步骤参考: YOLOv5 Weights & Biases Tutorial.
安装pip install wandb, 并在wandb官网上注册好账户后,先登录wandb
wandb login
然后执行YOLOv5训练程序
python train.py --img 640 --batch-size 32 --epochs 3 --data coco.yaml --weights yolov5s.pt --optimizer Adam --workers 6 --name yolov5s_20220404



网友评论