Merge pull request yunjey#25 from DingKe/master

yunjey · web-flow · commit 6b5af8730949 · 2017-04-23T22:58:03.000+09:00
use buit-in detach
diff --git a/tutorials/08 - Language Model/main-gpu.py b/tutorials/08 - Language Model/main-gpu.py
@@ -61,7 +61,7 @@ def forward(self, x, h):
 
 # Truncated Backpropagation 
 def detach(states):
-    return [Variable(state.data) for state in states] 
+    return [state.detach() for state in states] 
 
 # Training
 for epoch in range(num_epochs):
@@ -119,4 +119,4 @@ def detach(states):
             print('Sampled [%d/%d] words and save to %s'%(i+1, num_samples, sample_path))
 
 # Save the Trained Model
-torch.save(model.state_dict(), 'model.pkl')
+torch.save(model.state_dict(), 'model.pkl')
diff --git a/tutorials/08 - Language Model/main.py b/tutorials/08 - Language Model/main.py
@@ -61,7 +61,7 @@ def forward(self, x, h):
 
 # Truncated Backpropagation 
 def detach(states):
-    return [Variable(state.data) for state in states] 
+    return [state.detach() for state in states] 
 
 # Training
 for epoch in range(num_epochs):
@@ -119,4 +119,4 @@ def detach(states):
             print('Sampled [%d/%d] words and save to %s'%(i+1, num_samples, sample_path))
 
 # Save the Trained Model
-torch.save(model.state_dict(), 'model.pkl')
+torch.save(model.state_dict(), 'model.pkl')
diff --git a/tutorials/10 - Generative Adversarial Network/main-gpu.py b/tutorials/10 - Generative Adversarial Network/main-gpu.py
@@ -77,7 +77,7 @@ def forward(self, x):
         
         noise = Variable(torch.randn(images.size(0), 128)).cuda()
         fake_images = generator(noise)
-        outputs = discriminator(fake_images) 
+        outputs = discriminator(fake_images.detach()) 
         fake_loss = criterion(outputs, fake_labels)
         fake_score = outputs
         
@@ -107,4 +107,4 @@ def forward(self, x):
 
 # Save the Models 
 torch.save(generator.state_dict(), './generator.pkl')
-torch.save(discriminator.state_dict(), './discriminator.pkl')
+torch.save(discriminator.state_dict(), './discriminator.pkl')
diff --git a/tutorials/10 - Generative Adversarial Network/main.py b/tutorials/10 - Generative Adversarial Network/main.py
@@ -77,7 +77,7 @@ def forward(self, x):
         
         noise = Variable(torch.randn(images.size(0), 128))
         fake_images = generator(noise)
-        outputs = discriminator(fake_images) 
+        outputs = discriminator(fake_images.detach()) 
         fake_loss = criterion(outputs, fake_labels)
         fake_score = outputs
         
@@ -107,4 +107,4 @@ def forward(self, x):
 
 # Save the Models 
 torch.save(generator.state_dict(), './generator.pkl')
-torch.save(discriminator.state_dict(), './discriminator.pkl')
+torch.save(discriminator.state_dict(), './discriminator.pkl')
diff --git a/tutorials/11 - Deep Convolutional Generative Adversarial Network/main-gpu.py b/tutorials/11 - Deep Convolutional Generative Adversarial Network/main-gpu.py
@@ -102,7 +102,7 @@ def forward(self, x):
         
         noise = Variable(torch.randn(images.size(0), 128)).cuda()
         fake_images = generator(noise)
-        outputs = discriminator(fake_images) 
+        outputs = discriminator(fake_images.detach()) 
         fake_loss = criterion(outputs, fake_labels)
         fake_score = outputs
         
@@ -131,4 +131,4 @@ def forward(self, x):
 
 # Save the Models 
 torch.save(generator.state_dict(), './generator.pkl')
-torch.save(discriminator.state_dict(), './discriminator.pkl')
+torch.save(discriminator.state_dict(), './discriminator.pkl')
diff --git a/tutorials/11 - Deep Convolutional Generative Adversarial Network/main.py b/tutorials/11 - Deep Convolutional Generative Adversarial Network/main.py
@@ -102,7 +102,7 @@ def forward(self, x):
         
         noise = Variable(torch.randn(images.size(0), 128))
         fake_images = generator(noise)
-        outputs = discriminator(fake_images) 
+        outputs = discriminator(fake_images.detch()) 
         fake_loss = criterion(outputs, fake_labels)
         fake_score = outputs
         
@@ -131,4 +131,4 @@ def forward(self, x):
 
 # Save the Models
 torch.save(generator.state_dict(), './generator.pkl')
-torch.save(discriminator.state_dict(), './discriminator.pkl')
+torch.save(discriminator.state_dict(), './discriminator.pkl')
