4
0
mirror of https://github.com/nerfstudio-project/nerfstudio.git synced 2026-11-11 18:44:21 +08:00

added method to argument list

This commit is contained in:
evonneng 2022-05-30 15:56:28 +00:00
parent b7020873f7
commit a782306c6b

View File

@ -1,5 +1,8 @@
#!/bin/bash
METHOD=$1
shift
# Deal with gpu's. If passed in, use those.
GPU_IDX=("$@")
if [ -z "$GPU_IDX" ]; then
@ -35,6 +38,7 @@ for dataset in ${DATASETS[@]}; do
((idx=idx+1))
fi
python scripts/run_train.py \
--config-name ${METHOD} \
'~logging.writer.LocalWriter' \
data.dataset.data_directory=data/blender/${dataset} \
data.dataset.downscale_factor=1 \
@ -43,5 +47,5 @@ for dataset in ${DATASETS[@]}; do
graph.steps_per_save=25000 \
graph.max_num_iterations=2000000 \
logging.enable_profiler=False &
echo "Launched ${dataset} on gpu ${GPU_IDX[$idx]}, ${tag}"
echo "Launched ${METHOD} ${dataset} on gpu ${GPU_IDX[$idx]}, ${tag}"
done