vector-quantize-pytorch icon indicating copy to clipboard operation
vector-quantize-pytorch copied to clipboard

[Feature request] Implement dead code revival

Open jordiae opened this issue 4 years ago • 7 comments
trafficstars

jordiae avatar Oct 02 '21 17:10 jordiae

hey! do you have a link to the paper with this technique?

lucidrains avatar Oct 05 '21 14:10 lucidrains

hey! do you have a link to the paper with this technique?

Hey, sorry about the lack of details, I wanted to add more context but completely forgot about it! I'm not really sure about which was the first paper that used this technique, but I'd say is this one: https://arxiv.org/pdf/2005.00341.pdf

jordiae avatar Oct 06 '21 11:10 jordiae

@jordiae the new improved vqgan has some tips too! https://openreview.net/forum?id=pfNyExj7z2 section 3.2 Screen Shot 2021-10-06 at 7 12 12 PM

lucidrains avatar Oct 07 '21 02:10 lucidrains

@jordiae let me implement these two tweaks from the improved VQGAN paper, then i can look at the jukebox method

lucidrains avatar Oct 07 '21 02:10 lucidrains

@jordiae Hey Jordi! Implemented both of the proposals from the VQGan paper in 0.3.1 and 0.3.2 - let me know if they work for you. I'll also try implementing the dead code revival technique from the soundstream paper, brought up by Wes in the other issue

lucidrains avatar Oct 19 '21 00:10 lucidrains

@jordiae Hey Jordi! Implemented both of the proposals from the VQGan paper in 0.3.1 and 0.3.2 - let me know if they work for you. I'll also try implementing the dead code revival technique from the soundstream paper, brought up by Wes in the other issue

Cool! Thanks!

jordiae avatar Oct 19 '21 17:10 jordiae

@jordiae can't guarantee it is :bug: free though :rofl: lemme know how it fares

lucidrains avatar Oct 19 '21 17:10 lucidrains