mirror of
https://github.com/nerfstudio-project/nerfstudio.git
synced 2026-11-13 03:14:00 +08:00
added logic for printing things (#309)
* added logic for printing things * dynamic viewer url
This commit is contained in:
1 parent
7a084eaf62
commit
e96f7c0617
4 files changed
+45
-17
No files matched your search
@@ -83,14 +83,19 @@ class Trainer:
|
||||
self.pipeline: Pipeline
|
||||
self.optimizers: Optimizers
|
||||
self.start_step = 0
|
||||
# logging variables
|
||||
writer.setup_event_writers(config.logging, max_iter=config.trainer.max_num_iterations)
|
||||
profiler.setup_profiler(config.logging)
|
||||
# visualizer variable
|
||||
banner_messages = None
|
||||
self.visualizer_state = viewer_utils.VisualizerState(config.viewer)
|
||||
if config.viewer.enable:
|
||||
banner_messages = [f"Viewer at: {self.visualizer_state.viewer_url}"]
|
||||
self.grad_scaler = GradScaler(enabled=self.mixed_precision)
|
||||
# training callbacks
|
||||
self.callbacks: List[Callback]
|
||||
# logging variables
|
||||
writer.setup_event_writers(
|
||||
config.logging, max_iter=config.trainer.max_num_iterations, banner_messages=banner_messages
|
||||
)
|
||||
profiler.setup_profiler(config.logging)
|
||||
|
||||
def setup(self, test_mode=False):
|
||||
"""Setup the Trainer by calling other setup functions.
|
||||
|
||||
Reference in new issue
Block a user