TastyPancakes/Flux.jl
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Merge pull request #805 from DrChainsaw/prefor-so-fix

Browse Source 
Fix for #803
This commit is contained in:
Mike J Innes 2019-07-11 16:02:54 +01:00 committed by GitHub
commit 174adf94d9
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 11 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
src/treelike.jl
View File

@ -31,6 +31,7 @@ end
function prefor(f, x; seen = IdSet()) function prefor(f, x; seen = IdSet())
x ∈ seen && return x ∈ seen && return
push!(seen, x)
f(x) f(x)
foreach(x -> prefor(f, x, seen = seen), children(x)) foreach(x -> prefor(f, x, seen = seen), children(x))
return return

10
test/utils.jl
View File

@ -85,6 +85,16 @@ end
@test size.(params(m)) == [(5, 10), (5,)] @test size.(params(m)) == [(5, 10), (5,)]
m = RNN(10, 5) m = RNN(10, 5)
@test size.(params(m)) == [(5, 10), (5, 5), (5,), (5,)] @test size.(params(m)) == [(5, 10), (5, 5), (5,), (5,)]
# Layer duplicated in same chain, params just once pls.
c = Chain(m, m)
@test size.(params(c)) == [(5, 10), (5, 5), (5,), (5,)]
# Self-referential array. Just want params, no stack overflow pls.
r = Any[nothing,m]
Flux.children(a::Vector{Any}) = Tuple(a)
r[1] = r
@test size.(params(r)) == [(5, 10), (5, 5), (5,), (5,)]
end end
@testset "Basic Stacking" begin @testset "Basic Stacking" begin