updating scripts; compatible with pipeline refactor (#278)

This commit is contained in:
evonneng authored and GitHub committed 2022-08-15 13:26:37 -07:00
1 parent 387057cd61
commit efd5d39652
5 files changed
+10 -6

No files matched your search

+2
View File
@@ -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)