Skip to content

Commit d08072a

Browse files
committed
Updated docstring examples for loadmodel! to cover more cases
1 parent a59f688 commit d08072a

File tree

1 file changed

+8
-13
lines changed

1 file changed

+8
-13
lines changed

src/loading.jl

Lines changed: 8 additions & 13 deletions
Original file line numberDiff line numberDiff line change
@@ -41,28 +41,23 @@ Zero bias vectors and `bias=false` are considered equivalent
4141
4242
# Examples
4343
```julia
44-
julia> using Flux: loadmodel!
45-
46-
julia> dst = Chain(Dense(Flux.ones32(2, 5)), Dense(2 => 1))
44+
julia> dst = Chain(Dense(Flux.ones32(2, 5, tanh)), Dense(2 => 1; bias = [1f0]))
4745
Chain(
48-
Dense(5 => 2), # 12 parameters
46+
Dense(5 => 2, tanh), # 12 parameters
4947
Dense(2 => 1), # 3 parameters
5048
) # Total: 4 arrays, 15 parameters, 316 bytes.
5149
52-
julia> src = Chain(Dense(5 => 2), Dense(2 => 1));
53-
54-
julia> all(isone, dst[1].weight)
50+
julia> dst[1].weight ≈ ones(2, 5) # by construction
5551
true
5652
57-
julia> loadmodel!(dst, src);
53+
julia> src = Chain(Dense(5 => 2, relu), Dense(2 => 1, bias=false));
5854
59-
julia> dst[1].weight ≈ ones(2, 5)
60-
false
55+
julia> Flux.loadmodel!(dst, src);
6156
62-
julia> dst[1].weight == src[1].weight
63-
true
57+
julia> dst[1].weight ≈ ones(2, 5) # values changed
58+
false
6459
65-
julia> dst[2].bias == src[2].bias
60+
julia> iszero(dst[2].bias)
6661
true
6762
```
6863

0 commit comments

Comments
 (0)