FasterTransformer icon indicating copy to clipboard operation
FasterTransformer copied to clipboard
verified publisher icon NVIDIA

How to get a npz file that satisfy the input requirement?

Open jy00161yang opened this issue 2 years ago • 1 comments

Branch/Tag/Commit

main

Docker Image Version

nvcr.io

GPU name

T4

CUDA Driver

515

Reproduced Steps

Is there any example of showing how to transfer a pytorch model to npz file that satisfy the vit input while using tensorrt plugin?

jy00161yang avatar Aug 30 '23 10:08 jy00161yang

@jy00161yang Did u find any solutions?

proevgenii avatar Sep 20 '23 16:09 proevgenii