L-electron-Rare/mlx
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Fix repr of conv layers (#3275)

Browse Source
This commit is contained in:
Angelos Katharopoulos
2026-03-18 22:47:38 -07:00
committed by GitHub
parent dbfbc0f65a
commit 45af0df90b
1 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files
python/mlx/nn/layers/convolution.py
+3 -3
View File
@@ -66,7 +66,7 @@ class Conv1d(Module):
def _extra_repr(self):
return (
f"{self.weight.shape[-1]}, {self.weight.shape[0]}, "
f"{self.weight.shape[-1] * self.groups}, {self.weight.shape[0]}, "
f"kernel_size={self.weight.shape[1]}, stride={self.stride}, "
f"padding={self.padding}, dilation={self.dilation}, "
f"groups={self.groups}, "
@@ -146,7 +146,7 @@ class Conv2d(Module):
def _extra_repr(self):
return (
f"{self.weight.shape[-1]}, {self.weight.shape[0]}, "
f"{self.weight.shape[-1] * self.groups}, {self.weight.shape[0]}, "
f"kernel_size={self.weight.shape[1:3]}, stride={self.stride}, "
f"padding={self.padding}, dilation={self.dilation}, "
f"groups={self.groups}, "
@@ -219,7 +219,7 @@ class Conv3d(Module):
def _extra_repr(self):
return (
f"{self.weight.shape[-1]}, {self.weight.shape[0]}, "
f"{self.weight.shape[-1] * self.groups}, {self.weight.shape[0]}, "
f"kernel_size={self.weight.shape[1:4]}, stride={self.stride}, "
f"padding={self.padding}, dilation={self.dilation}, "
f"bias={'bias' in self}"