Skip to content

Commit

Permalink
fix an error
Browse files Browse the repository at this point in the history
  • Loading branch information
lizhiqi49 committed Apr 28, 2024
1 parent cd3aa08 commit 84e74fb
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions custom/threestudio-3dgs/system/sugar_mvcontrol.py
Original file line number Diff line number Diff line change
Expand Up @@ -690,9 +690,9 @@ def training_step(self, batch, batch_idx):
)

if self.global_step % self.cfg.freq.render_normal == 0:
self._render_type == "normal"
self._render_type = "normal"
else:
self._render_type == "rgb"
self._render_type = "rgb"

out_2d = self.training_substep(batch, batch_idx, guidance="2d")
total_loss += out_2d["loss"]
Expand Down

0 comments on commit 84e74fb

Please sign in to comment.