tink/nerfstudio
4
0
Fork 0
mirror of https://github.com/nerfstudio-project/nerfstudio.git synced 2026-11-11 18:44:21 +08:00
Code

Check port before loading data (#1633)

Browse Source 
* Check port before loading data

* Fix formatting
This commit is contained in:
Alvin Xu 2023-03-22 14:37:35 -07:00 committed by GitHub
parent 3ec8c06e83
commit 9b99802042
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 12 additions and 11 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

23
nerfstudio/engine/trainer.py
View File

@ -138,6 +138,18 @@ class Trainer:
'test': loads train/test datasets into memory
'inference': does not load any dataset into memory
"""
# set up viewer if enabled
viewer_log_path = self.base_dir / self.config.viewer.relative_log_filename
self.viewer_state, banner_messages = None, None
if self.config.is_viewer_enabled() and self.local_rank == 0:
datapath = self.config.data
if datapath is None:
datapath = self.base_dir
self.viewer_state, banner_messages = viewer_utils.setup_viewer(
self.config.viewer, log_filename=viewer_log_path, datapath=datapath
)
self._check_viewer_warnings()
self.pipeline = self.config.pipeline.setup(
device=self.device, test_mode=test_mode, world_size=self.world_size, local_rank=self.local_rank
)
@ -153,17 +165,6 @@ class Trainer:
)
)
# set up viewer if enabled
viewer_log_path = self.base_dir / self.config.viewer.relative_log_filename
self.viewer_state, banner_messages = None, None
if self.config.is_viewer_enabled() and self.local_rank == 0:
datapath = self.pipeline.datamanager.get_datapath()
if datapath is None:
datapath = self.base_dir
self.viewer_state, banner_messages = viewer_utils.setup_viewer(
self.config.viewer, log_filename=viewer_log_path, datapath=datapath
)
self._check_viewer_warnings()
# set up writers/profilers if enabled
writer_log_path = self.base_dir / self.config.logging.relative_log_dir
writer.setup_event_writer(