WIP: GumbelSoftmax / RelaxedOneHotCategoricalStraightThrough - #562

Open

daydreamt wants to merge 14 commits into

pyro-ppl:masterfrom

daydreamt:RelaxedOneHotCategoricalStraightThrough

Commits on Mar 29, 2020

WIP: Add initial experiments based on pytorch's gumbel_softmax: need more reading on relaxed_categorical and transformations of distributions instead
daydreamt
committed
d95aaf8

Commits on Apr 5, 2020

Move GumbelSoftmaxProbs to continuous, because it needs access to Gumbel
daydreamt
committed
3a847af

Commits on Apr 6, 2020

Commits on Apr 7, 2020

Fix lint problems
daydreamt
committed
78451d6
Make gumbel_soft_max_probs behavior consistent in preparation of passing the shape tests;
daydreamt
committed
8ee6ed0
Fix high/low temperature non-one hot tests
daydreamt
committed
7c6837b

Commits on May 3, 2020

Merge master into origin/RelaxedOneHotCategoricalStraightThrough
daydreamt
committed
fad229e

Commits on Jul 19, 2020

Merge master
daydreamt
committed
33b77a9