Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

error running model inference using tao-toolkit 5.0 #14

Open
statscol opened this issue Dec 12, 2023 · 0 comments
Open

error running model inference using tao-toolkit 5.0 #14

statscol opened this issue Dec 12, 2023 · 0 comments

Comments

@statscol
Copy link

statscol commented Dec 12, 2023

Hello,

Been trying to run inference using tao-toolkit 4.0(nvcr.io/nvidia/tao/tao-toolkit:4.0.0-deploy) for both FaceDetect and FaceDetectIR but whenever I try to generate the rtr engine file I get the following error

detectnet_v2 gen_trt_engine \
                    -m $ETLT_FILE_PATH \
                    -e LOCAL_PATH_TO/nvidia-tao/tao_deploy/specs/FaceDetect/FaceDetect_trt.txt \
                    -k $MODEL_KEY \
                    --data_type $DATA_TYPE \
                    --batch_size 1 \
                    --max_batch_size 1 \
                    --engine_file $OUT_FILEPATH
##Error
Traceback (most recent call last):
  File "/usr/local/lib/python3.8/dist-packages/google/protobuf/text_format.py", line 1721, in ParseFloat
    return float(text)
ValueError: could not convert string to float: '8.0e'
google.protobuf.text_format.ParseError: 30:22 : '    translate_max_y: 8.0e': Couldn't parse float: 8.0e

The file I'm using as the experiment_spec is the same FaceDetect_trt.txt in this repo.

Does anyone know any workaround? also tried tao-toolkit 5.0 and didn't work.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

1 participant