Use masking for backward operation

nadavbh12 / VQ-VAE

Minimalist implementation of VQ-VAE in Pytorch

BSD 3-Clause "New" or "Revised" License

508 stars 85 forks source link

Use masking for backward operation #4

Closed jadore801120 closed 6 years ago

jadore801120 commented 6 years ago

Hello, thanks for this awesome repo! Follow the TODO comment in the code, I rewrite the backward part with masking. Please take a look. Thanks. Yu-Hsiang

jadore801120 commented 6 years ago

Hi, thanks for the review! I did not notice that before. I have fixed it, please take a look.