Remove dataparser

This commit is contained in:
Matt Tancik committed 2023-02-27 12:36:01 -08:00
1 parent 57736cfafc
commit a83d443dc3
9 files changed
+124 -57

No files matched your search

+6 -2
View File
@@ -219,8 +219,12 @@ def main(config: TrainerConfig) -> None:
config.set_timestamp()
if config.data:
CONSOLE.log("Using --data alias for --data.pipeline.datamanager.dataparser.data")
config.pipeline.datamanager.dataparser.data = config.data
CONSOLE.log("Using --data alias for --data.pipeline.datamanager.data")
config.pipeline.datamanager.data = config.data
if config.prompt:
CONSOLE.log("Using --prompt alias for --data.pipeline.model.prompt")
config.pipeline.model.prompt = config.prompt
if config.load_config:
CONSOLE.log(f"Loading pre-set config from: {config.load_config}")
+3 -3
View File
@@ -76,16 +76,16 @@ class RunViewer:
)
while True:
viewer_state.vis["renderingState/isTraining"].write(False)
self._update_viewer_state(viewer_state, config, pipeline)
self._update_viewer_state(viewer_state, pipeline)
def _update_viewer_state(self, viewer_state: viewer_utils.ViewerState, config: TrainerConfig, pipeline: Pipeline):
def _update_viewer_state(self, viewer_state: viewer_utils.ViewerState, pipeline: Pipeline):
"""Updates the viewer state by rendering out scene with current pipeline
Returns the time taken to render scene.
"""
# NOTE: step must be > 0 otherwise the rendering would not happen
step = 1
num_rays_per_batch = config.pipeline.datamanager.train_num_rays_per_batch
num_rays_per_batch = pipeline.datamanager.get_train_rays_per_batch()
with TimeWriter(writer, EventName.ITER_VIS_TIME) as _:
try:
viewer_state.update_scene(self, step, pipeline.model, num_rays_per_batch)