update readme

This commit is contained in:
evonneng committed 2022-05-11 17:59:35 +00:00
1 parent 6a77ca89aa
commit 9671ec3c33
2 files changed
+20

No files matched your search

+14
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