update for new version of torch

Author: morvanzhou

.gitignore

@@ -0,0 +1 @@
+.idea

tutorial-contents/406_GAN.py

@@ -55,24 +55,22 @@ def artist_works():     # painting from the famous artist (real target)
 plt.ion()   # something about continuous plotting
 
 for step in range(10000):
-    artist_paintings = artist_works()           # real painting from artist
-    G_ideas = torch.randn(BATCH_SIZE, N_IDEAS)  # random ideas
+    artist_paintings = artist_works()  # real painting from artist
+    G_ideas = torch.randn(BATCH_SIZE, N_IDEAS, requires_grad=True)  # random ideas\n
     G_paintings = G(G_ideas)                    # fake painting from G (random ideas)
-
-    prob_artist0 = D(artist_paintings)          # D try to increase this prob
     prob_artist1 = D(G_paintings)               # D try to reduce this prob
-
+    G_loss = torch.mean(torch.log(1. - prob_artist1))  
+    opt_G.zero_grad()
+    G_loss.backward()
+    opt_G.step()
+     
+    prob_artist0 = D(artist_paintings)          # D try to increase this prob
+    prob_artist1 = D(G_paintings.detach())  # D try to reduce this prob
     D_loss = - torch.mean(torch.log(prob_artist0) + torch.log(1. - prob_artist1))
-    G_loss = torch.mean(torch.log(1. - prob_artist1))
-
     opt_D.zero_grad()
     D_loss.backward(retain_graph=True)      # reusing computational graph
     opt_D.step()
 
-    opt_G.zero_grad()
-    G_loss.backward()
-    opt_G.step()
-
     if step % 50 == 0:  # plotting
         plt.cla()
         plt.plot(PAINT_POINTS[0], G_paintings.data.numpy()[0], c='#4AD631', lw=3, label='Generated painting',)