mirror of
https://github.com/nerfstudio-project/nerfstudio.git
synced 2026-11-15 04:13:57 +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
@@ -222,7 +222,6 @@ class VanillaDataloader(Dataloader): # pylint: disable=abstract-method
|
||||
@profiler.time_function
|
||||
def setup_dataloader(config: DataloaderConfig, device: str, test_mode=False) -> Dataloader:
|
||||
"""Setup the dataloader."""
|
||||
|
||||
dataset_train = instantiate_from_dict_config(config.train_dataset)
|
||||
train_datasetinputs = dataset_train.get_dataset_inputs(split="train")
|
||||
if config.eval_dataset is not None:
|
||||
|
||||
@@ -71,7 +71,7 @@ class DataloaderConfig:
|
||||
train_num_rays_per_batch: int = MISSING
|
||||
train_num_images_to_sample_from: int = MISSING
|
||||
eval_dataset: Optional[Dict[str, Any]] = None
|
||||
eval_image_indices: List[int] = MISSING
|
||||
eval_image_indices: Optional[List[int]] = None
|
||||
eval_num_rays_per_chunk: int = MISSING
|
||||
|
||||
|
||||
@@ -141,7 +141,11 @@ def setup_config(config: DictConfig) -> Config:
|
||||
trainer = TrainerConfig(**config.trainer)
|
||||
experiment_name = config.experiment_name
|
||||
method_name = config.method_name
|
||||
pipeline = PipelineConfig(**config.pipeline)
|
||||
dataloader = DataloaderConfig(**config.pipeline.dataloader)
|
||||
model = ModelConfig(**config.pipeline.model)
|
||||
pipeline = PipelineConfig(
|
||||
_target_=config.pipeline._target_, dataloader=dataloader, model=model # pylint:disable=protected-access
|
||||
)
|
||||
optimizers = config.optimizers
|
||||
viewer = ViewerConfig(**config.viewer)
|
||||
|
||||
|
||||
Reference in new issue
Block a user