mmdeploy
mmdeploy copied to clipboard
[Feature] 问一句,mask2former如何能不使用mmdeploy::grid_sampler而使用torch.grid_sampler
Motivation
mm这边转mask2former一直是使用自定义的算子mmdeploy::grid_sampler,但是trt fp16没有实现只有fp32,如何才能让模型使用torch.grid_sampler
Related resources
No response
Additional context
No response
我也想问,转onnx用的grid_sampler是mmdeploy自带的算子。。根本没有用啊,这种第三方算子实际生产根本不适用
我已解决
老铁,请教一下怎么解决的,需要修改哪里
老铁,请教一下怎么解决的,需要修改哪里
你用的tensor engine吧,我不是用的这个,我只需要转onnx
我也需要先转onnx再转trt,咱们步骤还是有一些重合的,请问应该怎么修改,感激不尽
我也需要先转onnx再转trt,咱们步骤还是有一些重合的,请问应该怎么修改,感激不尽
有点忘记了,你需要把mmdeploy关于grid sampler的symbolic operator禁用掉