@@ -41,28 +41,23 @@ Zero bias vectors and `bias=false` are considered equivalent
41
41
42
42
# Examples
43
43
```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]))
47
45
Chain(
48
- Dense(5 => 2), # 12 parameters
46
+ Dense(5 => 2, tanh), # 12 parameters
49
47
Dense(2 => 1), # 3 parameters
50
48
) # Total: 4 arrays, 15 parameters, 316 bytes.
51
49
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
55
51
true
56
52
57
- julia> loadmodel!(dst, src );
53
+ julia> src = Chain(Dense(5 => 2, relu), Dense(2 => 1, bias=false) );
58
54
59
- julia> dst[1].weight ≈ ones(2, 5)
60
- false
55
+ julia> Flux.loadmodel!(dst, src);
61
56
62
- julia> dst[1].weight == src[1].weight
63
- true
57
+ julia> dst[1].weight ≈ ones(2, 5) # values changed
58
+ false
64
59
65
- julia> dst[2].bias == src[2].bias
60
+ julia> iszero( dst[2].bias)
66
61
true
67
62
```
68
63
0 commit comments