We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent cf2e674 commit 7f9c382Copy full SHA for 7f9c382
src/flows/neuralspline.jl
@@ -27,7 +27,6 @@ function NeuralSplineCoupling(
27
# output dim of the NN
28
output_dims = (3K - 1)*num_of_transformed_dims
29
# one big mlp that outputs all the knots and derivatives for all the transformed dimensions
30
- # todo: ensure type stability
31
nn = fnn(input_dims, hdims, output_dims; output_activation=nothing, paramtype=paramtype)
32
33
mask = Bijectors.PartitionMask(dim, mask_idx)
0 commit comments