mirror of
https://github.com/nerfstudio-project/nerfstudio.git
synced 2026-11-11 10:34:29 +08:00
14 lines
488 B
Bash
Executable File
14 lines
488 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} \
|
|
experiment_name=blender_${dataset} \
|
|
logging.enable_stats=False \
|
|
logging.enable_profiler=False &
|
|
echo "Launched ${dataset} on gpu ${!GPU_IDX}"
|
|
GPU_IDX=$((GPU_IDX+1))
|
|
done |