tink/nerfstudio
4
0
Fork 0
mirror of https://github.com/nerfstudio-project/nerfstudio.git synced 2026-11-11 02:24:38 +08:00
Code

update readme

Browse Source
This commit is contained in:
evonneng 2022-05-11 17:59:35 +00:00
parent 6a77ca89aa
commit 9671ec3c33
2 changed files with 20 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
README.md
View File

@ -105,3 +105,9 @@ pip install py-spy
```
ipython --pdb scripts/run_train.py
```
5. **Benchmarking** For launching training jobs automatically on blender dataset
```
./scripts/benchmarking/launch_train_blender.sh <gpu0> <gpu1> ... <gpu7>
```

14
scripts/benchmarking/launch_train_blender.sh Executable file
View File

@ -0,0 +1,14 @@
#!/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