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

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