Unverified Commit adc0516d authored by bryandlee's avatar bryandlee Committed by GitHub

Update README.md

parent 8314798e
...@@ -15,7 +15,7 @@ python convert_weights.py ...@@ -15,7 +15,7 @@ python convert_weights.py
**Inference** **Inference**
``` ```
python test.py --input_dir [image_folder_path] python test.py --input_dir [image_folder_path] --device [cpu/cuda]
``` ```
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment