File size: 2,079 Bytes
c165cd8 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 |
#!/bin/bash
# outdoor
EXPERIMENT_PREFIX=360_v2_multiscale
SCENE=("bicycle" "garden" "stump" )
DATA_ROOT=/SSD_DISK/datasets/360_v2
len=${#SCENE[@]}
for((i=0; i<$len; i++ ))
do
EXPERIMENT=$EXPERIMENT_PREFIX/"${SCENE[i]}"
DATA_DIR="$DATA_ROOT"/"${SCENE[i]}"
accelerate launch train.py \
--gin_configs=configs/multi360.gin \
--gin_bindings="Config.data_dir = '${DATA_DIR}'" \
--gin_bindings="Config.exp_name = '${EXPERIMENT}'" \
--gin_bindings="Config.factor = 4"
accelerate launch eval.py \
--gin_configs=configs/multi360.gin \
--gin_bindings="Config.data_dir = '${DATA_DIR}'" \
--gin_bindings="Config.exp_name = '${EXPERIMENT}'" \
--gin_bindings="Config.factor = 4"
accelerate launch render.py \
--gin_configs=configs/360.gin \
--gin_bindings="Config.data_dir = '${DATA_DIR}'" \
--gin_bindings="Config.exp_name = '${EXPERIMENT}'" \
--gin_bindings="Config.render_path = True" \
--gin_bindings="Config.render_path_frames = 120" \
--gin_bindings="Config.render_video_fps = 30" \
--gin_bindings="Config.factor = 4"
done
# indoor "Config.factor = 2"
SCENE=("room" "counter" "kitchen" "bonsai")
len=${#SCENE[@]}
for((i=0; i<$len; i++ ))
do
EXPERIMENT=$EXPERIMENT_PREFIX/"${SCENE[i]}"
DATA_DIR="$DATA_ROOT"/"${SCENE[i]}"
accelerate launch train.py \
--gin_configs=configs/multi360.gin \
--gin_bindings="Config.data_dir = '${DATA_DIR}'" \
--gin_bindings="Config.exp_name = '${EXPERIMENT}'" \
--gin_bindings="Config.factor = 2"
accelerate launch eval.py \
--gin_configs=configs/multi360.gin \
--gin_bindings="Config.data_dir = '${DATA_DIR}'" \
--gin_bindings="Config.exp_name = '${EXPERIMENT}'" \
--gin_bindings="Config.factor = 2"
accelerate launch render.py \
--gin_configs=configs/360.gin \
--gin_bindings="Config.data_dir = '${DATA_DIR}'" \
--gin_bindings="Config.exp_name = '${EXPERIMENT}'" \
--gin_bindings="Config.render_path = True" \
--gin_bindings="Config.render_path_frames = 120" \
--gin_bindings="Config.render_video_fps = 30" \
--gin_bindings="Config.factor = 2"
done |