Fix a bug for GPU

network/network_utils.py

@@ -12,7 +12,7 @@ import numpy as np
 class BaseNet:
     def set_gpu(self, gpu):
         if gpu >= 0 and torch.cuda.is_available():
-            self.device = torch.device('gpu:%d' % (gpu))
+            self.device = torch.device('cuda:%d' % (gpu))
         else:
             self.device = torch.device('cpu')
         self.to(self.device)