Skip to content

Commit 16d6333

Browse files
committed
Remove init from base encoder and decoder
1 parent 1e9b169 commit 16d6333

2 files changed

Lines changed: 0 additions & 10 deletions

File tree

src/auto_cast/decoders/base.py

Lines changed: 0 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -9,11 +9,6 @@
99
class Decoder(nn.Module, ABC):
1010
"""Base Decoder."""
1111

12-
def __init__(self, latent_dim: int, output_channels: int) -> None:
13-
super().__init__()
14-
self.latent_dim = latent_dim
15-
self.output_channels = output_channels
16-
1712
def decode(self, z: Tensor) -> Tensor:
1813
"""Decode the latent tensor back to the original space.
1914

src/auto_cast/encoders/base.py

Lines changed: 0 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -9,11 +9,6 @@
99
class Encoder(nn.Module, ABC):
1010
"""Base encoder."""
1111

12-
def __init__(self, latent_dim: int, input_channels: int) -> None:
13-
super().__init__()
14-
self.latent_dim = latent_dim
15-
self.input_channels = input_channels
16-
1712
def encode(self, x: Tensor) -> Tensor:
1813
"""Encode the input tensor into the latent space.
1914

0 commit comments

Comments
 (0)