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