credit assignment

Author: lucidrains

README.md

@@ -464,3 +464,13 @@ $ accelerate launch train.py
     volume  = {abs/2210.13438}
 }
 ```
+
+```bibtex
+@article{Hu2017SqueezeandExcitationN,
+    title   = {Squeeze-and-Excitation Networks},
+    author  = {Jie Hu and Li Shen and Gang Sun},
+    journal = {2018 IEEE/CVF Conference on Computer Vision and Pattern Recognition},
+    year    = {2017},
+    pages   = {7132-7141}
+}
+```

audiolm_pytorch/soundstream.py

@@ -130,6 +130,7 @@ def forward(self, x, return_intermediates = False):
         return out, intermediates
 
 # autoregressive squeeze excitation
+# https://arxiv.org/abs/1709.01507
 
 class SqueezeExcite(nn.Module):
     def __init__(self, dim, reduction_factor = 4, dim_minimum = 8):