diff --git a/models/autoencoder.py b/models/autoencoder.py new file mode 100644 index 0000000..b66fb73 --- /dev/null +++ b/models/autoencoder.py @@ -0,0 +1,18 @@ +import torch +import torch.nn as nn + + +class Encoder(nn.Module): + def __init__(self, input_size, hidden_size, output_size): + super(Encoder, self).__init__() + + def forward(self, x: torch.Tensor) -> torch.Tensor: + pass + + +class Decoder(nn.Module): + def __init__(self): + super(Decoder, self).__init__() + + def forward(self, x: torch.Tensor) -> torch.Tensor: + pass