mirror of
https://github.com/nerfstudio-project/nerfstudio.git
synced 2026-11-12 02:45:34 +08:00
updating scripts; compatible with pipeline refactor (#278)
This commit is contained in:
1 parent
387057cd61
commit
efd5d39652
5 files changed
+10
-6
No files matched your search
@@ -20,6 +20,7 @@ from nerfactory.cameras.camera_paths import (
|
||||
)
|
||||
from nerfactory.pipelines.base import Pipeline, setup_pipeline
|
||||
from nerfactory.utils import io as io_utils
|
||||
from nerfactory.utils.config import setup_config
|
||||
from nerfactory.utils.misc import human_format
|
||||
from nerfactory.utils.writer import TimeWriter
|
||||
|
||||
@@ -164,6 +165,7 @@ def main(
|
||||
|
||||
config.trainer.resume_train.load_dir = checkpoint_dir
|
||||
config.pipeline.dataloader.eval_image_indices = None
|
||||
config = setup_config(config) # converting to typed config
|
||||
|
||||
device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
|
||||
# setup pipeline (which includes the dataloaders)
|
||||
|
||||
Reference in new issue
Block a user