From 332942e5f9f53c2ed79db90161a7b2c059b67322 Mon Sep 17 00:00:00 2001 From: NASEEM A P Date: Mon, 2 Oct 2023 12:41:33 +0530 Subject: [PATCH] added example for training from from pre-trained weight --- README.md | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/README.md b/README.md index 737e536..74da8f8 100644 --- a/README.md +++ b/README.md @@ -122,6 +122,10 @@ You can train your **YOLO-NAS** model with **Single Command Line** **Example:** ``` python3 train.py --data /dir/dataset/data.yaml --batch 6 --epoch 100 --model yolo_nas_m --size 640 + +# From Pre-trained weight +python3 train.py --data /dir/dataset/data.yaml --batch 6 --epoch 100 --model yolo_nas_m --size 640 \ + --weight runs/train2/ckpt_latest.pth ``` ### If your training ends in 65th epoch (total 100 epochs), now you can start from 65th epoch and complete your 100 epochs training. **Example:**