set gpu_id in all codes to 0

LynnHo · Feb 26, 2019 · ed45429 · ed45429
1 parent 4c5daad
commit ed45429
Show file tree

Hide file tree

Showing 12 changed files with 12 additions and 12 deletions.
diff --git a/train_cartoon_dcgan.py b/train_cartoon_dcgan.py
@@ -15,7 +15,7 @@
 batch_size = 64
 lr = 0.0002
 z_dim = 100
-gpu_id = 3
+gpu_id = 0
 
 ''' data '''
 # you should prepare your own data in ./data/faces

diff --git a/train_cartoon_lsgan.py b/train_cartoon_lsgan.py
@@ -15,7 +15,7 @@
 batch_size = 64
 lr = 0.0002
 z_dim = 100
-gpu_id = 1
+gpu_id = 0
 
 ''' data '''
 # you should prepare your own data in ./data/faces

diff --git a/train_cartoon_wgan.py b/train_cartoon_wgan.py
@@ -17,7 +17,7 @@
 z_dim = 100
 clip = 0.01
 n_critic = 5
-gpu_id = 3
+gpu_id = 0
 
 ''' data '''
 # you should prepare your own data in ./data/faces

diff --git a/train_cartoon_wgan_gp.py b/train_cartoon_wgan_gp.py
@@ -16,7 +16,7 @@
 lr = 0.0002
 z_dim = 100
 n_critic = 5
-gpu_id = 1
+gpu_id = 0
 
 ''' data '''
 # you should prepare your own data in ./data/faces

diff --git a/train_celeba_dcgan.py b/train_celeba_dcgan.py
@@ -15,7 +15,7 @@
 batch_size = 64
 lr = 0.0002
 z_dim = 100
-gpu_id = 3
+gpu_id = 0
 
 ''' data '''
 # you should prepare your own data in ./data/img_align_celeba

diff --git a/train_celeba_lsgan.py b/train_celeba_lsgan.py
@@ -15,7 +15,7 @@
 batch_size = 64
 lr = 0.0002
 z_dim = 100
-gpu_id = 3
+gpu_id = 0
 
 ''' data '''
 # you should prepare your own data in ./data/img_align_celeba

diff --git a/train_celeba_wgan.py b/train_celeba_wgan.py
@@ -17,7 +17,7 @@
 z_dim = 100
 clip = 0.01
 n_critic = 5
-gpu_id = 3
+gpu_id = 0
 
 ''' data '''
 # you should prepare your own data in ./data/img_align_celeba

diff --git a/train_celeba_wgan_gp.py b/train_celeba_wgan_gp.py
@@ -16,7 +16,7 @@
 lr = 0.0002
 z_dim = 100
 n_critic = 5
-gpu_id = 3
+gpu_id = 0
 
 ''' data '''
 # you should prepare your own data in ./data/img_align_celeba

diff --git a/train_mnist_dcgan.py b/train_mnist_dcgan.py
@@ -15,7 +15,7 @@
 batch_size = 64
 lr = 0.0002
 z_dim = 100
-gpu_id = 3
+gpu_id = 0
 
 ''' data '''
 utils.mkdir('./data/mnist/')

diff --git a/train_mnist_lsgan.py b/train_mnist_lsgan.py
@@ -15,7 +15,7 @@
 batch_size = 64
 lr = 0.0002
 z_dim = 100
-gpu_id = 3
+gpu_id = 0
 
 ''' data '''
 utils.mkdir('./data/mnist/')

diff --git a/train_mnist_wgan.py b/train_mnist_wgan.py
@@ -17,7 +17,7 @@
 z_dim = 100
 clip = 0.01
 n_critic = 5
-gpu_id = 3
+gpu_id = 0
 
 ''' data '''
 utils.mkdir('./data/mnist/')

diff --git a/train_mnist_wgan_gp.py b/train_mnist_wgan_gp.py
@@ -16,7 +16,7 @@
 lr = 0.0002
 z_dim = 100
 n_critic = 5
-gpu_id = 3
+gpu_id = 0
 
 ''' data '''
 utils.mkdir('./data/mnist/')