SW777SW commited on
Commit
10d3a0c
·
verified ·
1 Parent(s): 332b4d8

Update run_train_infer.sh

Browse files
Files changed (1) hide show
  1. run_train_infer.sh +6 -3
run_train_infer.sh CHANGED
@@ -37,14 +37,16 @@ for DATASET in "${DATASETS[@]}"; do
37
  MODEL_PATH=./output/infer/${DATASET}/${SCENE}/${N_VIEW}_views_${gs_train_iter}Iter_${pose_lr}PoseLR/
38
 
39
  # # ----- (1) Dust3r_coarse_geometric_initialization -----
40
- CMD_D1="CUDA_VISIBLE_DEVICES=${GPU_ID} python -W ignore ./coarse_init_infer.py \
 
41
  --img_base_path ${SOURCE_PATH} \
42
  --n_views ${N_VIEW} \
43
  --focal_avg \
44
  "
45
 
46
  # # ----- (2) Train: jointly optimize pose -----
47
- CMD_T="CUDA_VISIBLE_DEVICES=${GPU_ID} python -W ignore ./train_joint.py \
 
48
  -s ${SOURCE_PATH} \
49
  -m ${MODEL_PATH} \
50
  --n_views ${N_VIEW} \
@@ -54,7 +56,8 @@ for DATASET in "${DATASETS[@]}"; do
54
  "
55
 
56
  # ----- (3) Render interpolated pose & output video -----
57
- CMD_RI="CUDA_VISIBLE_DEVICES=${GPU_ID} python -W ignore ./render_by_interp.py \
 
58
  -s ${SOURCE_PATH} \
59
  -m ${MODEL_PATH} \
60
  --n_views ${N_VIEW} \
 
37
  MODEL_PATH=./output/infer/${DATASET}/${SCENE}/${N_VIEW}_views_${gs_train_iter}Iter_${pose_lr}PoseLR/
38
 
39
  # # ----- (1) Dust3r_coarse_geometric_initialization -----
40
+ # CMD_D1="CUDA_VISIBLE_DEVICES=${GPU_ID} python -W ignore ./coarse_init_infer.py \
41
+ CMD_D1="python -W ignore ./coarse_init_infer.py \
42
  --img_base_path ${SOURCE_PATH} \
43
  --n_views ${N_VIEW} \
44
  --focal_avg \
45
  "
46
 
47
  # # ----- (2) Train: jointly optimize pose -----
48
+ # CMD_T="CUDA_VISIBLE_DEVICES=${GPU_ID} python -W ignore ./train_joint.py \
49
+ CMD_T="python -W ignore ./train_joint.py \
50
  -s ${SOURCE_PATH} \
51
  -m ${MODEL_PATH} \
52
  --n_views ${N_VIEW} \
 
56
  "
57
 
58
  # ----- (3) Render interpolated pose & output video -----
59
+ # CMD_RI="CUDA_VISIBLE_DEVICES=${GPU_ID} python -W ignore ./render_by_interp.py \
60
+ CMD_RI="python -W ignore ./render_by_interp.py \
61
  -s ${SOURCE_PATH} \
62
  -m ${MODEL_PATH} \
63
  --n_views ${N_VIEW} \