HollrayChan issues

Repositories
Issues
Comments

Results 2 issues of


                                            HollrayChan

In fp16 mode, slice the result of vit, but the result is nan

Hi, I want to use my vit in fasttransformer, at the end of the network, I want to directly output some features but not al. so I implemented a cuda...

How convert weights from pytorch vit to tensorrt plugin？

Torch:1.10.2+cu113 TRT: 8.2.1.8 I train model with vit, I want to convert weights from pytorch vit to tensorrt. I wrote a function to transform the weights to adapt to the...