Skip to content

Commit f1aaf23

Browse files
authored
Merge pull request #121 from Oneflow-Inc/fix_readme_arg
update model to weights in README
2 parents 3547c58 + 0f60afa commit f1aaf23

File tree

1 file changed

+4
-4
lines changed

1 file changed

+4
-4
lines changed

README.md

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -89,16 +89,16 @@ YOLOv5实例分割模型支持使用 `--data coco128-seg.yaml` 参数自动下
8989

9090
```shell
9191
# Single-GPU
92-
python segment/train.py --model yolov5s-seg.of --data coco128-seg.yaml --epochs 5 --img 640
92+
python segment/train.py --weights yolov5s-seg.of --data coco128-seg.yaml --epochs 5 --img 640
9393

9494
# Multi-GPU DDP
95-
python -m oneflow.distributed.launch --nproc_per_node 4 segment/train.py --model yolov5s-seg.of --data coco128-seg.yaml --epochs 5 --img 640 --device 0,1,2,3
95+
python -m oneflow.distributed.launch --nproc_per_node 4 segment/train.py --weights yolov5s-seg.of --data coco128-seg.yaml --epochs 5 --img 640 --device 0,1,2,3
9696
```
9797

9898
注意 :
9999
- {`.of`: 代表OneFlow预训练权重 , `.pt`: 代表 PyTorch 预训练权重 }
100-
- `--model yolov5s-seg.of` 表示使用OneFlow预训练权重 , 也是支持使用 PyTorch 预训练权重 如 `--model yolov5s-seg.pt`
101-
- 模型权重将自动从 github 下载(*建议如果没有设置代理,可以提前将模型下载到电脑本地 使用 `--model 本地路径/yolov5s-seg.of`*)
100+
- `--weights yolov5s-seg.of` 表示使用OneFlow预训练权重 , 也是支持使用 PyTorch 预训练权重 如 `--weights yolov5s-seg.pt`
101+
- 模型权重将自动从 github 下载(*建议如果没有设置代理,可以提前将模型下载到电脑本地 使用 `--weights 本地路径/yolov5s-seg.of`*)
102102

103103
### val
104104

0 commit comments

Comments
 (0)