added training resume to README

naseemap47 · Sep 24, 2023 · 73c0e39 · 73c0e39
1 parent e8c2f1a
commit 73c0e39
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/README.md b/README.md
@@ -134,6 +134,12 @@ You can train your **YOLO-NAS** model with **Single Command Line**
 ```
 python3 train.py --data /dir/dataset/data.yaml --batch 6 --epoch 100 --model yolo_nas_m --size 640
 ```
+### If your training ends in 65th epoch (total 100 epochs), now you can start from 65th epoch and complete your 100 epochs training.
+**Example:**
+```
+python3 train.py --data /dir/dataset/data.yaml --batch 6 --epoch 100 --model yolo_nas_m --size 640 \
+                 --weight runs/train2/ckpt_latest.pth --resume
+```
 
 ## 📺 Inference
 You can Inference your **YOLO-NAS** model with **Single Command Line**