File tree Expand file tree Collapse file tree 1 file changed +7
-3
lines changed Expand file tree Collapse file tree 1 file changed +7
-3
lines changed Original file line number Diff line number Diff line change @@ -154,15 +154,19 @@ def __init__(
154154
155155 if in_channels % 2 != 0 :
156156 raise ValueError (
157- f"'in_channels' ({ in_channels } ) must be divisible by 'groups' ({ groups } )."
157+ "'in_channels' must be divisible by 'groups'."
158+ f"Found: in_channels={ in_channels } , groups={ groups } ."
158159 )
159160 if out_channels % 2 != 0 :
160161 raise ValueError (
161- f"'out_channels' ({ out_channels } ) must be divisible by 'groups' ({ groups } )."
162+ "'out_channels' must be divisible by 'groups'."
163+ f"Found: out_channels={ out_channels } , groups={ groups } ."
162164 )
163165
164166 kernel_size = to_ntuple (kernel_size , ndim )
165- self .weight = nn .Parameter (torch .randn (out_channels , in_channels // groups , * kernel_size ))
167+ self .weight = nn .Parameter (
168+ torch .randn (out_channels , in_channels // groups , * kernel_size )
169+ )
166170 self .bias = nn .Parameter (torch .randn (out_channels ,)) if bias else None
167171
168172 def forward (self , signal ):
You can’t perform that action at this time.
0 commit comments