Update README.md

adc0516d · bryandlee · GitHub · 8314798e · adc0516d
Unverified Commit adc0516d authored Feb 22, 2021 by bryandlee Committed by GitHub Feb 22, 2021
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

README.md README.md +1 -1

No files found.
--- 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]
 ```