hylee commited on
Commit
a7397f6
·
1 Parent(s): 0116024
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
- for str_id in str_ids:
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
- for str_id in str_ids:
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