search

yangxy / GPEN

2.39k stars 449 forks source link

TensorRT deployment #175

Open bychen7 opened 1 year ago

bychen7 commented 1 year ago

Hi, @yangxy Thank you for your open source code. I successfully deployed this model using TensorRT. Here is the link to my implementation: https://github.com/bychen7/Face-Restoration-TensorRT.

zimenglan-sysu-512 commented 1 year ago

hi @bychen7 can u provide the code of converting torch model to onnx file? thanks.

luhairong11 commented 10 months ago

I have successfully converted to onnx, but when converting to tensorrt, I get an error. [TRT] [E] /ecd1/ecd1.0/ecd1.0.0/Conv: two inputs (data and weights) are allowed only in explicit-quantization mode. In node 52 (parseGraph): INVALID_NODE: Invalid Node - /ecd1/ecd1.0/ecd1.0.0/Conv /ecd1/ecd1.0/ecd1.0.0/Conv: two inputs (data and weights) are allowed only in explicit-quantization mode.

bychen7 commented 10 months ago

Hi, @zimenglan-sysu-512 @luhairong11 You can refer to this.