Spaces:
Sleeping
Sleeping
Update options/base_options.py
Browse files- options/base_options.py +1 -1
options/base_options.py
CHANGED
|
@@ -22,7 +22,7 @@ class BaseOptions():
|
|
| 22 |
# basic parameters
|
| 23 |
parser.add_argument('--dataroot', required=True, help='path to images (should have subfolders trainA, trainB, valA, valB, etc)')
|
| 24 |
parser.add_argument('--name', type=str, default='experiment_name', help='name of the experiment. It decides where to store samples and models')
|
| 25 |
-
parser.add_argument('--gpu_ids', type=str, default='
|
| 26 |
parser.add_argument('--checkpoints_dir', type=str, default='./checkpoints', help='models are saved here')
|
| 27 |
# model parameters
|
| 28 |
parser.add_argument('--model', type=str, default='cycle_gan', help='chooses which model to use. [cycle_gan | pix2pix | test | colorization]')
|
|
|
|
| 22 |
# basic parameters
|
| 23 |
parser.add_argument('--dataroot', required=True, help='path to images (should have subfolders trainA, trainB, valA, valB, etc)')
|
| 24 |
parser.add_argument('--name', type=str, default='experiment_name', help='name of the experiment. It decides where to store samples and models')
|
| 25 |
+
parser.add_argument('--gpu_ids', type=str, default='-1', help='gpu ids: e.g. 0 0,1,2, 0,2. use -1 for CPU')
|
| 26 |
parser.add_argument('--checkpoints_dir', type=str, default='./checkpoints', help='models are saved here')
|
| 27 |
# model parameters
|
| 28 |
parser.add_argument('--model', type=str, default='cycle_gan', help='chooses which model to use. [cycle_gan | pix2pix | test | colorization]')
|