edward2 icon indicating copy to clipboard operation
edward2 copied to clipboard

Published 20 hours ago • verified publisher icon google

HeteroscedasticSNGPLayer error with num_classes=2

Open Krasner opened this issue 2 years ago • 0 comments

HeteroscedasticSNGPLayer inherits from MCSoftmaxDenseFA which has an assert check:

https://github.com/google/edward2/blob/2a852121f5e3c47c88e4212aae86d8ba33febdbe/edward2/tensorflow/layers/heteroscedastic.py#L605

but itself is written in a way that should accept num_classes=2: https://github.com/google/edward2/blob/2a852121f5e3c47c88e4212aae86d8ba33febdbe/edward2/tensorflow/layers/hetsngp.py#L232

Krasner avatar Feb 08 '23 15:02 Krasner