mirror of https://github.com/tracel-ai/burn.git
Fix tiny typo in mathematical expression (#2867)
This commit is contained in:
parent
a6b5210aee
commit
3b4974f011
|
@ -305,7 +305,7 @@ are set using the configuration of the corresponding neural network's underlying
|
|||
specific case, we have chosen to expand the tensor channels from 1 to 8 with the first layer, then
|
||||
from 8 to 16 with the second layer, using a kernel size of 3 on all dimensions. We also use the
|
||||
adaptive average pooling module to reduce the dimensionality of the images to an 8 by 8 matrix,
|
||||
which we will flatten in the forward pass to have a 1024 (16 _ 8 _ 8) resulting tensor.
|
||||
which we will flatten in the forward pass to have a 1024 (16 * 8 * 8) resulting tensor.
|
||||
|
||||
Now let's see how the forward pass is defined.
|
||||
|
||||
|
|
Loading…
Reference in New Issue