Implementation of a straight-through gradient wrapper to allow for discrete latent representations. Provides binary discretizer which maps hidden representations to {0, 1} and a learnable multi-value discretizer, which maps hidden activations to their closest value in a set of given size.
-
Notifications
You must be signed in to change notification settings - Fork 0
Implementation of a straight-through gradient wrapper to allow for discrete latent representations. Provides binary discretizer which maps hidden representations to {0, 1} and a learnable multi-value discretizer, which maps hidden activations to their closest value in a set of given size.
LGirrbach/Discrete-Latent-Representation
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Folders and files
| Name | Name | Last commit message | Last commit date | |
|---|---|---|---|---|
Repository files navigation
About
Implementation of a straight-through gradient wrapper to allow for discrete latent representations. Provides binary discretizer which maps hidden representations to {0, 1} and a learnable multi-value discretizer, which maps hidden activations to their closest value in a set of given size.