Finegrain/refiners
1
0
Fork 0
mirror of https://github.com/finegrain-ai/refiners.git synced 2024-11-22 06:08:46 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

remove unused Conv1d

Browse source
This commit is contained in:
Pierre Chapuis 2024-01-29 16:36:17 +01:00 committed by Cédric Deltheil
parent 86867e9318
commit 07954a55ab
1 changed files with 0 additions and 30 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

30
src/refiners/fluxion/layers/conv.py
View file

@ -34,36 +34,6 @@ class Conv2d(nn.Conv2d, WeightedModule):
self.use_bias = use_bias self.use_bias = use_bias
class Conv1d(nn.Conv1d, WeightedModule):
def __init__(
self,
in_channels: int,
out_channels: int,
kernel_size: int | tuple[int],
stride: int | tuple[int] = 1,
padding: int | tuple[int] | str = 0,
groups: int = 1,
use_bias: bool = True,
dilation: int | tuple[int] = 1,
padding_mode: str = "zeros",
device: Device | str | None = None,
dtype: DType | None = None,
) -> None:
super().__init__( # type: ignore
in_channels,
out_channels,
kernel_size,
stride,
padding,
dilation,
groups,
use_bias,
padding_mode,
device,
dtype,
)
class ConvTranspose2d(nn.ConvTranspose2d, WeightedModule): class ConvTranspose2d(nn.ConvTranspose2d, WeightedModule):
def __init__( def __init__(
self, self,