From 2ffaee0c0df4277e0eabbca4ca57879bd574b19c Mon Sep 17 00:00:00 2001 From: Markus Enzweiler <76874238+menzHSE@users.noreply.github.com> Date: Wed, 6 Dec 2023 16:17:58 +0100 Subject: [PATCH] Updated default argument for stride to 1 in Conv2d() in the docstring (#22) --- python/mlx/nn/layers/convolution.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/python/mlx/nn/layers/convolution.py b/python/mlx/nn/layers/convolution.py index c9b6ce363..c6928e188 100644 --- a/python/mlx/nn/layers/convolution.py +++ b/python/mlx/nn/layers/convolution.py @@ -78,7 +78,7 @@ class Conv2d(Module): out_channels (int): The number of output channels. kernel_size (int or tuple): The size of the convolution filters. stride (int or tuple, optional): The size of the stride when - applying the filter. Default: 0. + applying the filter. Default: 1. padding (int or tuple, optional): How many positions to 0-pad the input with. Default: 0. bias (bool, optional): If ``True`` add a learnable bias to the