Skip to content

Commit 5753150

Browse files
committed
change Any[] to Flux.Dense[]
1 parent dc73605 commit 5753150

File tree

2 files changed

+2
-1
lines changed

2 files changed

+2
-1
lines changed

example/utils.jl

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,5 @@
11
using Random, Distributions, LinearAlgebra
2+
using Bijectors
23
using Bijectors: transformed
34

45
function compare_trained_and_untrained_flow(

src/flows/utils.jl

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -78,7 +78,7 @@ function fnn(
7878
) where {T<:AbstractFloat}
7979
# Create a chain of dense layers
8080
# First layer (need to use Any[] otherwise it complains about type instability when pushing layers)
81-
layers = Any[Flux.Dense(input_dim, hidden_dims[1], inlayer_activation)]
81+
layers = Flux.Dense[Flux.Dense(input_dim, hidden_dims[1], inlayer_activation)]
8282

8383
# Hidden layers
8484
for i in 1:(length(hidden_dims) - 1)

0 commit comments

Comments
 (0)