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

Update docs/src/models/basics.md 

Co-Authored-By: Carlo Lucibello <carlo.lucibello@gmail.com>
This commit is contained in:
Dhairya Gandhi 2020-02-12 11:26:11 +05:30 committed by GitHub
parent ee6d950696
commit d5ed9a4478
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
docs/src/models/basics.md
View File

@ -231,7 +231,7 @@ Flux.trainable(a::Affine) = (a.W, a.b,)
Only the fields returned by `trainable` will be collected as trainable parameters of the layer when calling `Flux.params`. Only the fields returned by `trainable` will be collected as trainable parameters of the layer when calling `Flux.params`.
Another way of achieving this is through the `@functor` macro. Here, wee can mark the fields we are interested in like so: Another way of achieving this is through the `@functor` macro. Here, wee can mark the fields we are interested in like so:
Another way of achieving this is through the `@functor` macro. Here, we can mark the fields we are interested in by grouping them in the second argument:
```julia ```julia
Flux.@functor Affine (W,) Flux.@functor Affine (W,)
``` ```