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 -1
View File
@@ -63,7 +63,7 @@ for dataset in ${DATASETS[@]}; do
--checkpoint-dir ${checkpoint_dir} \
--rendered-output-name rgb \
--output-filename ${output_dir}/${base_config_name}/blender_${dataset}_${month}-${date}-${year}_${seconds}.json \
--config-overrides data.dataset_inputs_train.data_directory=data/blender/${dataset} data.dataset_inputs_eval.data_directory=data/blender/${dataset} &
--config-overrides pipeline.dataloader.train_dataset.data_directory=data/blender/${dataset} &
echo "Launched ${checkpoint_dir} on gpu ${GPU_IDX[$idx]}"
# update gpu
+1 -2
View File
@@ -55,8 +55,7 @@ for dataset in ${DATASETS[@]}; do
python scripts/run_train.py \
--config-name ${config_name} \
'~logging.writer.LocalWriter' \
data.dataset_inputs_train.data_directory=data/blender/${dataset} \
data.dataset_inputs_eval.data_directory=data/blender/${dataset} \
pipeline.dataloader.train_dataset.data_directory=data/blender/${dataset} \
experiment_name=blender_${dataset}_${tag} \
trainer.model_dir=nerfactory_models/ \
trainer.steps_per_save=25000 \
+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)