feat: encoder skeleton code
This commit is contained in:
parent
c8f2f860e6
commit
3258da3015
1 changed files with 18 additions and 0 deletions
18
models/autoencoder.py
Normal file
18
models/autoencoder.py
Normal 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
|
||||
Reference in a new issue