rosinality / vq-vae-2-pytorch

Implementation of Generating Diverse High-Fidelity Images with VQ-VAE-2 in PyTorch
Other
1.62k stars 272 forks source link

weight_v meaning #35

Open Dashi-1997 opened 4 years ago

Dashi-1997 commented 4 years ago

Hello, can you explain what do the following codes mean? if self.causal > 0: self.conv.conv.weightv.data[:, :, -1, self.causal:].zero()

rosinality commented 4 years ago

weight_v is weight before of applying weight norm.