zhangyiss/mlx
1
0
Fork 0
mirror of https://github.com/ml-explore/mlx.git synced 2025-06-26 10:41:14 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

Updated default argument for stride to 1 in Conv2d() in the docstring (#22)

Browse Source
This commit is contained in:
Markus Enzweiler 2023-12-06 16:17:58 +01:00 committed by GitHub
parent 36b245b287
commit 2ffaee0c0d
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
python/mlx/nn/layers/convolution.py
View File

@ -78,7 +78,7 @@ class Conv2d(Module):
out_channels (int): The number of output channels. out_channels (int): The number of output channels.
kernel_size (int or tuple): The size of the convolution filters. kernel_size (int or tuple): The size of the convolution filters.
stride (int or tuple, optional): The size of the stride when 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 padding (int or tuple, optional): How many positions to 0-pad
the input with. Default: 0. the input with. Default: 0.
bias (bool, optional): If ``True`` add a learnable bias to the bias (bool, optional): If ``True`` add a learnable bias to the