From e68755c6aac34586cc1fc958f2802dbd8a3243d6 Mon Sep 17 00:00:00 2001 From: Essam Date: Fri, 24 May 2024 11:44:51 +0300 Subject: [PATCH] Update docs/src/interface/Custom Builders.md Co-authored-by: Anthony Blaom, PhD --- docs/src/interface/Custom Builders.md | 14 ++++++-------- 1 file changed, 6 insertions(+), 8 deletions(-) diff --git a/docs/src/interface/Custom Builders.md b/docs/src/interface/Custom Builders.md index 098cb8b0..5a3514c8 100644 --- a/docs/src/interface/Custom Builders.md +++ b/docs/src/interface/Custom Builders.md @@ -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`.