From 357921c226a44d890eb6e0135d51a5ea43d3821e Mon Sep 17 00:00:00 2001 From: NASEEM A P Date: Fri, 22 Sep 2023 10:36:21 +0530 Subject: [PATCH] added train args - resume --- README.md | 1 + 1 file changed, 1 insertion(+) diff --git a/README.md b/README.md index d8ce695..10c4638 100644 --- 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)
`--gpus`: Train on multiple gpus
`--cpu`: Train on CPU
+ `--resume`: To resume model training
**Other Training Parameters:**
`--warmup_mode`: Warmup Mode, eg: Linear Epoch Step