Browse Source

fix typo

main
bluestyle97 6 months ago
parent
commit
423b9e10b6
  1. 2
      src/model_mesh.py

2
src/model_mesh.py

@ -56,7 +56,7 @@ class MVRecon(pl.LightningModule):
if 'weight' in k: if 'weight' in k:
sd_fc[k.replace('net.', 'net_sdf.')] = -v[0:1] sd_fc[k.replace('net.', 'net_sdf.')] = -v[0:1]
else: else:
sd_fc[k.replace('net.', 'net_sdf.')] = 3.0 - v[0:1] sd_fc[k.replace('net.', 'net_sdf.')] = 10.0 - v[0:1]
sd_fc[k.replace('net.', 'net_rgb.')] = v[1:4] sd_fc[k.replace('net.', 'net_rgb.')] = v[1:4]
else: else:
sd_fc[k.replace('net.', 'net_sdf.')] = v sd_fc[k.replace('net.', 'net_sdf.')] = v

Loading…
Cancel
Save