yocabon commited on
Commit
b587bb3
·
1 Parent(s): d1c9c7b

fix init={} being modified inplace

Browse files
Files changed (1) hide show
  1. mast3r/cloud_opt/sparse_ga.py +2 -1
mast3r/cloud_opt/sparse_ga.py CHANGED
@@ -14,6 +14,7 @@ import os
14
  from collections import namedtuple
15
  from functools import lru_cache
16
  from scipy import sparse as sp
 
17
 
18
  from mast3r.utils.misc import mkdir_for, hash_md5
19
  from mast3r.cloud_opt.utils.losses import gamma_loss
@@ -166,7 +167,7 @@ def sparse_scene_optimizer(imgs, subsample, imsizes, pps, base_focals, core_dept
166
  init={}, device='cuda', dtype=torch.float32,
167
  matching_conf_thr=5., loss_dust3r_w=0.01,
168
  verbose=True, dbg=()):
169
-
170
  # extrinsic parameters
171
  vec0001 = torch.tensor((0, 0, 0, 1), dtype=dtype, device=device)
172
  quats = [nn.Parameter(vec0001.clone()) for _ in range(len(imgs))]
 
14
  from collections import namedtuple
15
  from functools import lru_cache
16
  from scipy import sparse as sp
17
+ import copy
18
 
19
  from mast3r.utils.misc import mkdir_for, hash_md5
20
  from mast3r.cloud_opt.utils.losses import gamma_loss
 
167
  init={}, device='cuda', dtype=torch.float32,
168
  matching_conf_thr=5., loss_dust3r_w=0.01,
169
  verbose=True, dbg=()):
170
+ init = copy.deepcopy(init)
171
  # extrinsic parameters
172
  vec0001 = torch.tensor((0, 0, 0, 1), dtype=dtype, device=device)
173
  quats = [nn.Parameter(vec0001.clone()) for _ in range(len(imgs))]