Skip to content

Commit

Permalink
fix
Browse files Browse the repository at this point in the history
  • Loading branch information
MrNeRF committed Sep 27, 2024
1 parent 7bbf30d commit 40dc7ac
Showing 1 changed file with 6 additions and 7 deletions.
13 changes: 6 additions & 7 deletions examples/simple_trainer_scaffold.py
Original file line number Diff line number Diff line change
Expand Up @@ -49,12 +49,12 @@ class Config:
# Disable viewer
disable_viewer: bool = False
# Path to the .pt files. If provide, it will skip training and run evaluation only.
# ckpt: Optional[List[str]] = None
ckpt: Optional[List[str]] = field(
default_factory=lambda: [
"/home/paja/projects/gsplat_fork/results/ckpts/ckpt_1999_rank0.pt"
]
)
ckpt: Optional[List[str]] = None
# ckpt: Optional[List[str]] = field(
# default_factory=lambda: [
# "/home/paja/projects/gsplat_fork/results/ckpts/ckpt_1999_rank0.pt"
# ]
# )
# Render trajectory path
render_traj_path: str = "ellipse"

Expand Down Expand Up @@ -729,7 +729,6 @@ def train(self):
sh_degree=None,
near_plane=cfg.near_plane,
far_plane=cfg.far_plane,
image_ids=image_ids,
render_mode="RGB+ED" if cfg.depth_loss else "RGB",
)
if renders.shape[-1] == 4:
Expand Down

0 comments on commit 40dc7ac

Please sign in to comment.