Skip to content

Commit 4d56e75

Browse files
Add note on custom behavior
1 parent a12569a commit 4d56e75

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

src/functor.jl

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -55,6 +55,8 @@ Given a model or specific layers from a model, create a `Params` object pointing
5555
5656
This can be used with [`gradient`](@ref), or as input to the [`Flux.train!`](@ref Flux.train!) function.
5757
58+
The behaviour of `params` on custom types can be customized using [`Functor.@functor`](@ref) or [`Flux.trainable`](@ref).
59+
5860
# Examples
5961
```jldoctest
6062
julia> params(Chain(Dense(ones(2,3))), softmax)

0 commit comments

Comments
 (0)