search

kevinzakka / recurrent-visual-attention

A PyTorch Implementation of "Recurrent Models of Visual Attention"
MIT License
468 stars 123 forks source link

The comment mistake #44

Open cyccchs opened 2 years ago

cyccchs commented 2 years ago

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) this mistake appears in many places