Skip to content

Commit

Permalink
Merge pull request #121 from Oneflow-Inc/fix_readme_arg
Browse files Browse the repository at this point in the history
update model to weights in README
  • Loading branch information
ccssu authored Mar 14, 2023
2 parents 3547c58 + 0f60afa commit f1aaf23
Showing 1 changed file with 4 additions and 4 deletions.
8 changes: 4 additions & 4 deletions README.md
Original file line number Diff line number Diff line change
Expand Up @@ -89,16 +89,16 @@ YOLOv5实例分割模型支持使用 `--data coco128-seg.yaml` 参数自动下

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

# Multi-GPU DDP
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
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
```

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

### val

Expand Down

0 comments on commit f1aaf23

Please sign in to comment.