This commit is contained in:
Morvan Zhou
2017-05-10 14:45:00 +10:00
parent dd4447e5be
commit f826b4c40a
2 changed files with 2 additions and 2 deletions

View File

@ -52,7 +52,7 @@ D = nn.Sequential( # Discriminator
nn.Linear(ART_COMPONENTS, 128), # receive art work either from the famous artist or a newbie like G
nn.ReLU(),
nn.Linear(128, 1),
nn.Sigmoid(), # tell the art work is made by artist or newbie G
nn.Sigmoid(), # tell the probability that the art work is made by artist
)
opt_D = torch.optim.Adam(D.parameters(), lr=LR_D)

View File

@ -55,7 +55,7 @@ D = nn.Sequential( # Discriminator
nn.Linear(ART_COMPONENTS+1, 128), # receive art work either from the famous artist or a newbie like G with label
nn.ReLU(),
nn.Linear(128, 1),
nn.Sigmoid(), # tell the art work is made by artist or newbie G
nn.Sigmoid(), # tell the probability that the art work is made by artist
)
opt_D = torch.optim.Adam(D.parameters(), lr=LR_D)