4
0
mirror of https://github.com/nerfstudio-project/nerfstudio.git synced 2026-11-11 02:24:38 +08:00

[WIP] added preset button controls; moved rendering into timing loop (#270)

added preset button controls; moved rendering into timing loop
This commit is contained in:
evonneng 2022-08-12 09:53:29 -07:00 committed by GitHub
parent 8e32e7863e
commit b03755d6ab
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 6 additions and 1 deletions

View File

@ -105,7 +105,6 @@ class Trainer:
# Note: if visualizer used, the rendering of the visualizer will be included in the iteration train time
with TimeWriter(writer, EventName.ITER_TRAIN_TIME, step=step) as t:
loss_metric_dict = self.train_iteration(step)
with TimeWriter(writer, EventName.ITER_VIS_TIME, step=step) as t:
self.visualizer_state.update_scene(step, self.pipeline.model)

View File

@ -213,6 +213,12 @@ export function RenderControls() {
set_fov(v);
},
},
// training speed
"Train Speed": buttonGroup({
'Fast': () => setControls({ min_resolution: 10, max_resolution: 10 }),
'Balanced': () => setControls({min_resolution: 50, max_resolution: 512 }),
'Slow': () => setControls({ min_resolution: 100, max_resolution: 1024 }),
}),
}),
[
isTraining,