cyccchs

The comment in module.py (line 21) says "x" is a 4D tensor of shape (B, H, W, C) but it's actually a 4d tensor of shape (B, C, H, W)...