Skip to content

Instantly share code, notes, and snippets.

@nbertagnolli
Created March 1, 2020 23:08
Show Gist options
  • Save nbertagnolli/3c95816322971c5991d71280063de0ed to your computer and use it in GitHub Desktop.
Save nbertagnolli/3c95816322971c5991d71280063de0ed to your computer and use it in GitHub Desktop.
A basic discriminator for a simple GAN
class Discriminator(nn.Module):
def __init__(self, input_length: int):
super(Discriminator, self).__init__()
self.dense = nn.Linear(int(input_length), 1);
self.activation = nn.Sigmoid()
def forward(self, x):
return self.activation(self.dense(x))
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment