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

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