Skip to content
Snippets Groups Projects
Unverified Commit e8dc3367 authored by Haonan Yu's avatar Haonan Yu Committed by GitHub
Browse files

add discrete VAE (#1372) 

* add discrete VAE

* also return z_mode for discrete vae

* add ST gumbel-softmax

* update

* add scheduler option for the temperature

* address comments
parent 770ac568
No related merge requests found
Hide whitespace changes
Inline Side-by-side
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment