Swin-Transformer icon indicating copy to clipboard operation
Swin-Transformer copied to clipboard
verified publisher icon microsoft

supported dynamic batch for torchscript trace

Open MenglingD opened this issue 3 years ago • 1 comments

The window_reverse does not support dynamic batch because it cast the first dimension of windows to integer.

MenglingD avatar Aug 29 '22 11:08 MenglingD

Hi @MenglingD, I'm also working on this model. Do you think this modification can also be applied to swin transformer v2? And have you tried to convert this model to TensorRT? Thx!

knwng avatar Nov 09 '22 08:11 knwng