K.C. Tung
Results
2
comments of
K.C. Tung
Any ETA on this problem? Without the fix, torch2trt won't work for many models I tried: Hugging Face vision transformer, swintransformer, ViViT...etc
My code works for other models. So is it because your model somehow has some intermediate tensors being mapped to CPU?