Skip to content

Commit

Permalink
Update docs/src/interface/Custom Builders.md
Browse files Browse the repository at this point in the history
Co-authored-by: Anthony Blaom, PhD <[email protected]>
  • Loading branch information
EssamWisam and ablaom authored May 24, 2024
1 parent ca78165 commit e68755c
Showing 1 changed file with 6 additions and 8 deletions.
14 changes: 6 additions & 8 deletions docs/src/interface/Custom Builders.md
Original file line number Diff line number Diff line change
Expand Up @@ -41,14 +41,12 @@ This method must return a `Flux.Chain` instance, `chain`, subject to the
following conditions:

- `chain(x)` must make sense:

- for any `x <: Array{<:AbstractFloat, 2}` of size `(n_in,
batch_size)` where `batch_size` is any integer (for use with one
of the first three model types); or

- for any `x <: Array{<:Float32, 4}` of size `(W, H, n_channels,
batch_size)`, where `(W, H) = n_in`, `n_channels` is 1 or 3, and
`batch_size` is any integer (for use with `ImageClassifier`)
- for any `x <: Array{<:AbstractFloat, 2}` of size `(n_in,
batch_size)` where `batch_size` is any integer (for use with one
of the first three model types); or
- for any `x <: Array{<:Float32, 4}` of size `(W, H, n_channels,
batch_size)`, where `(W, H) = n_in`, `n_channels` is 1 or 3, and
`batch_size` is any integer (for use with `ImageClassifier`)

- The object returned by `chain(x)` must be an `AbstractFloat` vector
of length `n_out`.
Expand Down

0 comments on commit e68755c

Please sign in to comment.