fedex/mt
Public Access
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

ae elbow work

Browse Source
This commit is contained in:
Jan Kowalczyk
2025-06-10 13:58:38 +02:00
parent 156b6d2ac1
commit d88719e718
4 changed files with 126 additions and 82 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
Deep-SAD-PyTorch/src/networks/main.py
View File

@@ -9,7 +9,7 @@ from .subter_LeNet_Split import SubTer_LeNet_Split, SubTer_LeNet_Split_Autoencod
from .vae import VariationalAutoencoder
def build_network(net_name, ae_net=None):
def build_network(net_name, ae_net=None, rep_dim=1024):
"""Builds the neural network."""
implemented_networks = (
@@ -46,7 +46,7 @@ def build_network(net_name, ae_net=None):
net = MNIST_LeNet()
if net_name == "subter_LeNet":
net = SubTer_LeNet()
net = SubTer_LeNet(rep_dim=rep_dim)
if net_name == "subter_LeNet_Split":
net = SubTer_LeNet_Split()