feat: encoder skeleton code

This commit is contained in:
Robin Meersman 2025-12-09 21:07:43 +01:00
parent c8f2f860e6
commit 3258da3015

18
models/autoencoder.py Normal file
View file

@ -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