mirror of
https://github.com/finegrain-ai/refiners.git
synced 2024-11-09 15:02:01 +00:00
remove unused Conv1d
This commit is contained in:
parent
86867e9318
commit
07954a55ab
|
@ -34,36 +34,6 @@ class Conv2d(nn.Conv2d, WeightedModule):
|
|||
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):
|
||||
def __init__(
|
||||
self,
|
||||
|
|
Loading…
Reference in a new issue