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
|