We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 1e9b169 commit 16d6333Copy full SHA for 16d6333
2 files changed
src/auto_cast/decoders/base.py
@@ -9,11 +9,6 @@
9
class Decoder(nn.Module, ABC):
10
"""Base Decoder."""
11
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
-
17
def decode(self, z: Tensor) -> Tensor:
18
"""Decode the latent tensor back to the original space.
19
src/auto_cast/encoders/base.py
class Encoder(nn.Module, ABC):
"""Base encoder."""
- def __init__(self, latent_dim: int, input_channels: int) -> None:
- self.input_channels = input_channels
def encode(self, x: Tensor) -> Tensor:
"""Encode the input tensor into the latent space.
0 commit comments