Update GetStart.md

yangzhegithub · Jan 16, 2022 · 1f4a99a · 1f4a99a
1 parent a706f52
commit 1f4a99a
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/docs/GetStart.md b/docs/GetStart.md
@@ -28,6 +28,7 @@ cd yolov5_obb
 
 
 **2. Train**
+
 2.1 Train with specified GPUs. (for example with GPU=3)
 
 ```shell
@@ -42,12 +43,12 @@ python -m torch.distributed.launch --nproc_per_node 4 train.py --device 0,1,2,3
 
 2.3 Train the orignal dataset demo.
 ```shell
-python train.py --data 'data/yolov5obb_demo.yaml' 
+python train.py --data 'data/yolov5obb_demo.yaml' --epochs 10 --batch-size 1 --img 1024 --device 0
 ```
 
 2.4 Train the splited dataset demo.
 ```shell
-python train.py --data 'data/yolov5obb_demo_split.yaml' 
+python train.py --data 'data/yolov5obb_demo_split.yaml' --epochs 10 --batch-size 2 --img 1024 --device 0
 ```
 
 # Inferenece with pretrained models. (Splited Dataset)