Open yassouali opened 4 years ago
The current version only works for attconv & attstem with stride == 1. The change is simple, we change the self.query_conv to have the desired stride and update the height and width after folding.
self.query_conv
stride
height
width
The current version only works for attconv & attstem with stride == 1. The change is simple, we change the
self.query_conv
to have the desiredstride
and update theheight
andwidth
after folding.