From f826b4c40a09c1e5579bc3fabef6a74184afa4d2 Mon Sep 17 00:00:00 2001 From: Morvan Zhou Date: Wed, 10 May 2017 14:45:00 +1000 Subject: [PATCH] update --- tutorial-contents/406_GAN.py | 2 +- tutorial-contents/406_conditional_GAN.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/tutorial-contents/406_GAN.py b/tutorial-contents/406_GAN.py index 5140d6d..2968b79 100644 --- a/tutorial-contents/406_GAN.py +++ b/tutorial-contents/406_GAN.py @@ -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) diff --git a/tutorial-contents/406_conditional_GAN.py b/tutorial-contents/406_conditional_GAN.py index a7db932..065f665 100644 --- a/tutorial-contents/406_conditional_GAN.py +++ b/tutorial-contents/406_conditional_GAN.py @@ -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)