mirror of
https://github.com/nerfstudio-project/nerfstudio.git
synced 2026-11-11 18:44:21 +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
@@ -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
|
||||
|
||||
@@ -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 \
|
||||
|
||||
@@ -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)
|
||||
|
||||
Reference in new issue
Block a user