Skip to content

Commit 1dfebd1

Browse files
committed
export nsf layer
1 parent e4fa67b commit 1dfebd1

File tree

1 file changed

+4
-2
lines changed

1 file changed

+4
-2
lines changed

src/NormalizingFlows.jl

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -129,9 +129,11 @@ end
129129
# interface of contructing common flow layers
130130
include("flows/utils.jl")
131131
include("flows/realnvp.jl")
132-
# include("flows/neuralspline.jl")
132+
include("flows/neuralspline.jl")
133133

134-
export RealNVP_layer, realnvp
134+
export create_flow
135+
export RealNVP_layer, realnvp, AffineCoupling
136+
export NeuralSplineLayer
135137

136138

137139
end

0 commit comments

Comments
 (0)