Commit f545bd1b authored by Phil Wang's avatar Phil Wang
Browse files

credit assignment

parent e4b66cf0

README.md

+10 −0
Original line number Diff line number Diff line
@@ -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

+1 −0
Original line number Diff line number Diff line
@@ -130,6 +130,7 @@ class MultiScaleDiscriminator(nn.Module): 
        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):