added train args - resume

naseemap47 · Sep 22, 2023 · 357921c · 357921c
1 parent 1f65418
commit 357921c
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/README.md b/README.md
@@ -116,6 +116,7 @@ You can train your **YOLO-NAS** model with **Single Command Line**
   `-w`, `--weight`: path to pre-trained model weight (default: `coco` weight) <br>
   `--gpus`: Train on multiple gpus <br>
   `--cpu`: Train on CPU <br>
+  `--resume`: To resume model training <br>
 
   **Other Training Parameters:**<br>
   `--warmup_mode`: Warmup Mode, eg: Linear Epoch Step <br>