fix

demo/gan_compression/dataset/data_loader.py

@@ -36,4 +36,4 @@ def create_data(cfgs, direction='AtoB', eval_mode=False):
 
 
 def create_eval_data(cfgs, direction='AtoB'):
-    return create_data(cfgs, eval_mode=True)
+    return create_data(cfgs, direction=direction, eval_mode=True)

demo/gan_compression/gan_compression.py

@@ -43,7 +43,7 @@ class gan_compression:
         num = self.cfgs.gpu_num
 
         use_gpu, use_parallel = False, False
-        if num == -1:
+        if num == 0:
             use_gpu = False
         else:
             use_gpu = True

demo/gan_compression/utils/get_args.py

@@ -44,7 +44,7 @@ class configs:
             default='resnet',
             help="generator network in supernet")
         parser.add_argument(
-            '--gpu_num', type=int, default='0', help='GPU number.')
+            '--gpu_num', type=int, default='1', help='GPU number.')
         ### data
         parser.add_argument(
             '--batch_size', type=int, default=1, help="Minbatch size")