steffen/ae_toolbox_torch
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Done: Latent Space Viz

Browse Source 
ToDo: Visualization for variational spaces
Trajectory Coloring
Post Processing
Metric
Slurm Skript
This commit is contained in:
Si11ium
2019-08-23 09:54:00 +02:00
parent 744c0c50b7
commit 1a0400d736
9 changed files with 159 additions and 76 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
networks/modules.py
View File

@ -131,13 +131,13 @@ class AvgDimPool(Module):
# Generators, Decoders, Encoders, Discriminators
class Discriminator(Module):
def __init__(self, latent_dim, dataParams, dropout=.0, activation=ReLU):
def __init__(self, latent_dim, features, dropout=.0, activation=ReLU):
super(Discriminator, self).__init__()
self.dataParams = dataParams
self.features = features
self.latent_dim = latent_dim
self.l1 = Linear(self.latent_dim, self.dataParams['features'] * 10)
self.l2 = Linear(self.dataParams['features'] * 10, self.dataParams['features'] * 20)
self.lout = Linear(self.dataParams['features'] * 20, 1)
self.l1 = Linear(self.latent_dim, self.features * 10)
self.l2 = Linear(self.features * 10, self.features * 20)
self.lout = Linear(self.features * 20, 1)
self.dropout = Dropout(dropout)
self.activation = activation()
self.sigmoid = Sigmoid()