Chu Văn An comments

Repositories
Issues
Comments

Results 2 comments of


                                            Chu Văn An

Error when running fine-tuned segmentation model on Triton Inference server : Trying to create tensor with negative dimension -23: [0, -23]

> I removed `self.task == segment` from autobackend.py here: > > and set `task` in tasks.py here: > > and got this: > > ``` > Traceback (most recent call...

Error when running fine-tuned segmentation model on Triton Inference server : Trying to create tensor with negative dimension -23: [0, -23]

Hi @atheer174 , I just fixed the error by passing the `data="your_config.yaml"`when predicting the model. The config contains `names` key representing your custom classes. For example my yaml config: ```...