fix: conflicts
This commit is contained in:
commit
b178c097d8
90 changed files with 2034 additions and 11145 deletions
18
src/models/autoencoder.py
Normal file
18
src/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