Add final fc (head) layer to trt engine, tested on cifar10 and ViT-B_16

[yuanjiechen]

Hello, thanks for providing this good inference library.
When I use ViT-trt modules, I try to add the fc (head) layer to tensorrt engine and do inference.
I tested the result on cifar10 with ViT-B_16 calibrated model (ft2).
Tensorrt version: 8.4.3.1
Cuda version: 11.6

[byshiue]

Hi, yuanjiechen. Thank you for the submission. But we don't plan to modify the example to run on specific task now because we hope to provide a common template which only contains the backbone. And different users can apply this backbone on different tasks like classification and segmentation.