Skip to content

Commit 2086404

Browse files
committed
Move and edit comment, add link
1 parent fd54428 commit 2086404

File tree

1 file changed

+5
-3
lines changed

1 file changed

+5
-3
lines changed

docs/src/models/advanced.md

Lines changed: 5 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -12,10 +12,10 @@ struct CustomModel
1212
end
1313

1414
function (m::CustomModel)(x)
15-
return m.chain(x) + x
16-
17-
# You can put arbitrary code here, but note that everything here will be differentiated.
15+
# Arbitrary code can go here, but note that everything will be differentiated.
1816
# Zygote does not allow some operations, like mutating arrays.
17+
18+
return m.chain(x) + x
1919
end
2020

2121
# Call @functor to allow for training. Described below in more detail.
@@ -30,6 +30,8 @@ model = CustomModel(chain)
3030
model(rand(10))
3131
```
3232

33+
For an intro to Flux and automatic differentiation, see this [tutorial](https://fluxml.ai/tutorials/2020/09/15/deep-learning-flux.html)
34+
3335
## Customising Parameter Collection for a Model
3436

3537
Taking reference from our example `Affine` layer from the [basics](basics.md#Building-Layers-1).

0 commit comments

Comments
 (0)