From adc0516db916279cb5db1ce76d8dfd7e9c69dbb0 Mon Sep 17 00:00:00 2001 From: bryandlee Date: Tue, 23 Feb 2021 07:47:03 +0900 Subject: [PATCH] Update README.md --- README.md | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/README.md b/README.md index d201ccd..623f778 100644 --- a/README.md +++ b/README.md @@ -15,7 +15,7 @@ python convert_weights.py **Inference** ``` -python test.py --input_dir [image_folder_path] +python test.py --input_dir [image_folder_path] --device [cpu/cuda] ```