diff --git a/demo/gan_compression/dataset/data_loader.py b/demo/gan_compression/dataset/data_loader.py index fa5dd4ec25af3efad283f0ea63f26b7108cdc9d8..c0085cb63194f2434e4a1c763637b0e5efb49180 100644 --- a/demo/gan_compression/dataset/data_loader.py +++ b/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) diff --git a/demo/gan_compression/gan_compression.py b/demo/gan_compression/gan_compression.py index 47f0681005de546c7f21ca6d73f09fa44f5af6b3..20ac434f3f51ef9a9b2179f08b6b98f31b2ba81f 100644 --- a/demo/gan_compression/gan_compression.py +++ b/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 diff --git a/demo/gan_compression/utils/get_args.py b/demo/gan_compression/utils/get_args.py index 8d3ebc9a7f7f6ef6729224625b08af632312ba3d..6310b80a6415417f7c6040cfff519c94253dbf89 100644 --- a/demo/gan_compression/utils/get_args.py +++ b/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")