Update README.md

README.md

@@ -71,7 +71,7 @@ source ~/pytorch_env.sh
 ```
 export HIP_VISIBLE_DEVICES=0
 
-python train.py --data data/citrus.yaml --cfg models/yolov5s.yaml  --weights weights/yolov5s.pt --epochs 150 --batch-size 4
+python3 train.py --batch 32 --data /work/home/acqpriwny1/cyh/datasets/MaskDataSet/data.yaml --cfg 'yolov5s.yaml' --weights '' --project 'run/train' --hyp 'data/hyps/hyp.scratch-high.yaml' --epochs 1000  2>&1 | tee  yolov5s.log
 ```
 
 ### 单节点多卡训练