How can i convert the weights to tensorrt engine file?

NVlabs / SegFormer

Official PyTorch implementation of SegFormer

https://arxiv.org/abs/2105.15203

Other

2.54k stars 353 forks source link

Open JiangWeiHn opened 2 years ago

JiangWeiHn commented 2 years ago

As the title says，I want use this model in the Nvidia xavier,so i want change the weights to tensorrt file.Could you help me. thanks a lot

GlingK commented 1 year ago

@JiangWeiHn did you finally find a way to convert the segformer model to tensorrt?