Global query: wrong input transformation

[shkarupa-alex]

Here https://github.com/NVlabs/GCVit/blob/main/models/gc_vit.py#L382 "x" is features in channel-last format [batch, height, width, channels]
You passing it thru some Conv2D layers (FeatExtract) which work with features in channel-first format ([batch, height, width, channels]). But instead of transpose operation you are making reshape with mixes channels and spatial axis.

[ahatamiz]

@shkarupa-alex Thanks for the comment. It will be addressed.