mirror of
https://github.com/nerfstudio-project/nerfstudio.git
synced 2026-11-11 02:24:38 +08:00
18 lines
665 B
Bash
Executable File
18 lines
665 B
Bash
Executable File
#!/bin/bash
|
|
|
|
DATASETS=("mic" "ficus" "chair" "hotdog" "materials" "drums" "ship" "lego")
|
|
GPU_IDX=1
|
|
for dataset in ${DATASETS[@]}; do
|
|
export CUDA_VISIBLE_DEVICES=${!GPU_IDX}
|
|
python scripts/run_train.py \
|
|
data.dataset.data_directory=data/blender/${dataset} \
|
|
data.dataset.downscale_factor=1 \
|
|
experiment_name=blender_${dataset} \
|
|
graph.model_dir=mattport_models/ \
|
|
graph.steps_per_save=25000 \
|
|
graph.max_num_iterations=2000000 \
|
|
logging.enable_stats=False \
|
|
logging.enable_profiler=False &
|
|
echo "Launched ${dataset} on gpu ${!GPU_IDX}"
|
|
GPU_IDX=$((GPU_IDX+1))
|
|
done |