fix bug

Author: L1aoXingyu

04-Convolutional Neural Network/convolution_network.py

@@ -76,6 +76,9 @@ def forward(self, x):
         if use_gpu:
             img = Variable(img).cuda()
             label = Variable(label).cuda()
+        else:
+            img = Variable(img)
+            label = Variable(label)
         # 向前传播
         out = model(img)
         loss = criterion(out, label)

05-Recurrent Neural Network/recurrent_network.py

@@ -70,6 +70,9 @@ def forward(self, x):
         if use_gpu:
             img = Variable(img).cuda()
             label = Variable(label).cuda()
+        else:
+            img = Variable(img)
+            label = Variable(label)
         # 向前传播
         out = model(img)
         loss = criterion(out, label)