Unable to use heteroscedastic layers in tf.keras
Open
- Dominant language
- Jupyter Notebook
- Stars
- 712
- Forks
- 77
- Avg merge
- 9h 39m
- Merged PRs (30d)
- 1
Description
`TypeError: Dimension value must be integer or None or have an __index__ method, got value '' with type ''`
```
hidden_layer = tf.keras.layers.Dense(16)
inputs = tf.keras.Input((224,224,3))
out = hidden_layer(inputs)
output_layer = ed.layers.MCSoftmaxDense(num_classes=2)
output_layer(out)
```
Contributor guide
Assessment
This issue has not been assessed yet.