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

an error about loading model by using triton #11

Open
littletree123 opened this issue Mar 18, 2021 · 2 comments
Open

an error about loading model by using triton #11

littletree123 opened this issue Mar 18, 2021 · 2 comments

Comments

@littletree123
Copy link

hi, i use triton's docker to run your model,and my step is

  1. cp centerface to model_repository/

  2. because of the network, i download the model from browser and run 'python3 change_dim.py'
    and the sha256sum of my centerface.onnx is 77e394b51108381b4c4f7b4baf1c64ca9f4aba73e5e803b2636419578913b5fe

  3. docker run --rm -p8000:8000 -p8001:8001 -p8002:8002 -v/path/to/model_repository:/models nvcr.io/nvidia/tritonserver:20.09-py3 tritonserver --model-repository=/models --model-control-mode=explicit --load-model centerface

and i got an error: model_repository_manager.cc:899] failed to load 'centerface' version 1: Invalid argument: model 'centerface', tensor '537': the model expects 4 dimensions (shape [1,1,-1,-1]) but the model configuration specifies 4 dimensions (shape [1,1,120,160])

i don't know what the problem is. Thank you so much!

@monjha
Copy link
Collaborator

monjha commented Mar 19, 2021

Hi,

Please run this application with deepstream docker :https://ngc.nvidia.com/catalog/containers/nvidia:deepstream/tags and use the 5.1-21.02-triton tag.

@ghost
Copy link

ghost commented May 8, 2021

@littletree123 were you able to get the model working with DeepStream 5.1-21.02 and Triton?

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

2 participants