init
Browse files
APDrawingGAN2/options/base_options.py
CHANGED
@@ -171,22 +171,12 @@ class BaseOptions():
|
|
171 |
self.print_options(opt)
|
172 |
|
173 |
# set gpu ids
|
174 |
-
str_ids = opt.gpu_ids.split(',')
|
175 |
opt.gpu_ids = []
|
176 |
-
|
177 |
-
id = int(str_id)
|
178 |
-
if id >= 0:
|
179 |
-
opt.gpu_ids.append(id)
|
180 |
-
if len(opt.gpu_ids) > 0:
|
181 |
-
torch.cuda.set_device(opt.gpu_ids[0])
|
182 |
|
183 |
# set gpu ids
|
184 |
-
str_ids = opt.gpu_ids_p.split(',')
|
185 |
opt.gpu_ids_p = []
|
186 |
-
|
187 |
-
id = int(str_id)
|
188 |
-
if id >= 0:
|
189 |
-
opt.gpu_ids_p.append(id)
|
190 |
|
191 |
self.opt = opt
|
192 |
return self.opt
|
|
|
171 |
self.print_options(opt)
|
172 |
|
173 |
# set gpu ids
|
|
|
174 |
opt.gpu_ids = []
|
175 |
+
|
|
|
|
|
|
|
|
|
|
|
176 |
|
177 |
# set gpu ids
|
|
|
178 |
opt.gpu_ids_p = []
|
179 |
+
|
|
|
|
|
|
|
180 |
|
181 |
self.opt = opt
|
182 |
return self.opt
|