Skip to content

Commit

Permalink
fix path
Browse files Browse the repository at this point in the history
  • Loading branch information
zeakey committed Nov 28, 2024
1 parent 7951619 commit 5569018
Show file tree
Hide file tree
Showing 9 changed files with 14 additions and 14 deletions.
4 changes: 2 additions & 2 deletions examples/benchmarks/basic.sh
Original file line number Diff line number Diff line change
Expand Up @@ -14,15 +14,15 @@ do
echo "Running $SCENE"

# train without eval
CUDA_VISIBLE_DEVICES=0 python simple_trainer.py default --eval_steps -1 --disable_viewer --data_factor $DATA_FACTOR \
CUDA_VISIBLE_DEVICES=0 python examples/simple_trainer.py default --eval_steps -1 --disable_viewer --data_factor $DATA_FACTOR \
--render_traj_path $RENDER_TRAJ_PATH \
--data_dir data/360_v2/$SCENE/ \
--result_dir $RESULT_DIR/$SCENE/

# run eval and render
for CKPT in $RESULT_DIR/$SCENE/ckpts/*;
do
CUDA_VISIBLE_DEVICES=0 python simple_trainer.py default --disable_viewer --data_factor $DATA_FACTOR \
CUDA_VISIBLE_DEVICES=0 python examples/simple_trainer.py default --disable_viewer --data_factor $DATA_FACTOR \
--render_traj_path $RENDER_TRAJ_PATH \
--data_dir data/360_v2/$SCENE/ \
--result_dir $RESULT_DIR/$SCENE/ \
Expand Down
4 changes: 2 additions & 2 deletions examples/benchmarks/basic_2dgs.sh
Original file line number Diff line number Diff line change
Expand Up @@ -13,15 +13,15 @@ do
echo "Running $SCENE"

# train without eval
CUDA_VISIBLE_DEVICES=0 python simple_trainer_2dgs.py --eval_steps -1 --disable_viewer --data_factor $DATA_FACTOR \
CUDA_VISIBLE_DEVICES=0 python examples/simple_trainer_2dgs.py --eval_steps -1 --disable_viewer --data_factor $DATA_FACTOR \
--model_type 2dgs \
--data_dir data/360_v2/$SCENE/ \
--result_dir $RESULT_DIR/$SCENE/

# run eval and render
for CKPT in $RESULT_DIR/$SCENE/ckpts/*;
do
CUDA_VISIBLE_DEVICES=0 python simple_trainer_2dgs.py --disable_viewer --data_factor $DATA_FACTOR \
CUDA_VISIBLE_DEVICES=0 python examples/simple_trainer_2dgs.py --disable_viewer --data_factor $DATA_FACTOR \
--model_type 2dgs \
--data_dir data/360_v2/$SCENE/ \
--result_dir $RESULT_DIR/$SCENE/ \
Expand Down
2 changes: 1 addition & 1 deletion examples/benchmarks/basic_4gpus.sh
Original file line number Diff line number Diff line change
Expand Up @@ -13,7 +13,7 @@ do
echo "Running $SCENE"

# train and eval at the last step (30000)
CUDA_VISIBLE_DEVICES=0,1,2,3 python simple_trainer.py default --eval_steps 30000 --disable_viewer --data_factor $DATA_FACTOR \
CUDA_VISIBLE_DEVICES=0,1,2,3 python examples/simple_trainer.py default --eval_steps 30000 --disable_viewer --data_factor $DATA_FACTOR \
# 4 GPUs is effectively 4x batch size so we scale down the steps by 4x as well.
# "--packed" reduces the data transfer between GPUs, which leads to faster training.
--steps_scaler 0.25 --packed \
Expand Down
2 changes: 1 addition & 1 deletion examples/benchmarks/bilarf/mcmc_bilarf.sh
Original file line number Diff line number Diff line change
Expand Up @@ -13,7 +13,7 @@ do
echo "Running $SCENE"

# train without eval
CUDA_VISIBLE_DEVICES=0 python simple_trainer.py mcmc --disable_viewer --data_factor $DATA_FACTOR \
CUDA_VISIBLE_DEVICES=0 python examples/simple_trainer.py mcmc --disable_viewer --data_factor $DATA_FACTOR \
--use_bilateral_grid \
--render_traj_path $RENDER_TRAJ_PATH \
--data_dir $SCENE_DIR/$SCENE/ \
Expand Down
4 changes: 2 additions & 2 deletions examples/benchmarks/compression/mcmc.sh
Original file line number Diff line number Diff line change
Expand Up @@ -30,13 +30,13 @@ do
echo "Running $SCENE"

# train without eval
CUDA_VISIBLE_DEVICES=0 python simple_trainer.py mcmc --eval_steps -1 --disable_viewer --data_factor $DATA_FACTOR \
CUDA_VISIBLE_DEVICES=0 python examples/simple_trainer.py mcmc --eval_steps -1 --disable_viewer --data_factor $DATA_FACTOR \
--strategy.cap-max $CAP_MAX \
--data_dir data/360_v2/$SCENE/ \
--result_dir $RESULT_DIR/$SCENE/

# eval: use vgg for lpips to align with other benchmarks
CUDA_VISIBLE_DEVICES=0 python simple_trainer.py mcmc --disable_viewer --data_factor $DATA_FACTOR \
CUDA_VISIBLE_DEVICES=0 python examples/simple_trainer.py mcmc --disable_viewer --data_factor $DATA_FACTOR \
--strategy.cap-max $CAP_MAX \
--data_dir data/360_v2/$SCENE/ \
--result_dir $RESULT_DIR/$SCENE/ \
Expand Down
4 changes: 2 additions & 2 deletions examples/benchmarks/compression/mcmc_tt.sh
Original file line number Diff line number Diff line change
Expand Up @@ -23,13 +23,13 @@ do
echo "Running $SCENE"

# train without eval
CUDA_VISIBLE_DEVICES=0 python simple_trainer.py mcmc --eval_steps -1 --disable_viewer --data_factor 1 \
CUDA_VISIBLE_DEVICES=0 python examples/simple_trainer.py mcmc --eval_steps -1 --disable_viewer --data_factor 1 \
--strategy.cap-max $CAP_MAX \
--data_dir $SCENE_DIR/$SCENE/ \
--result_dir $RESULT_DIR/$SCENE/

# eval: use vgg for lpips to align with other benchmarks
CUDA_VISIBLE_DEVICES=0 python simple_trainer.py mcmc --disable_viewer --data_factor 1 \
CUDA_VISIBLE_DEVICES=0 python examples/simple_trainer.py mcmc --disable_viewer --data_factor 1 \
--strategy.cap-max $CAP_MAX \
--data_dir $SCENE_DIR/$SCENE/ \
--result_dir $RESULT_DIR/$SCENE/ \
Expand Down
2 changes: 1 addition & 1 deletion examples/benchmarks/fisheye/mcmc_zipnerf.sh
Original file line number Diff line number Diff line change
Expand Up @@ -14,7 +14,7 @@ do
echo "Running $SCENE"

# train and eval
CUDA_VISIBLE_DEVICES=0 python simple_trainer.py mcmc --disable_viewer --data_factor $DATA_FACTOR \
CUDA_VISIBLE_DEVICES=0 python examples/simple_trainer.py mcmc --disable_viewer --data_factor $DATA_FACTOR \
--strategy.cap-max $CAP_MAX \
--opacity_reg 0.001 \
--use_bilateral_grid \
Expand Down
2 changes: 1 addition & 1 deletion examples/benchmarks/fisheye/mcmc_zipnerf_undistorted.sh
Original file line number Diff line number Diff line change
Expand Up @@ -14,7 +14,7 @@ do
echo "Running $SCENE"

# train and eval
CUDA_VISIBLE_DEVICES=0 python simple_trainer.py mcmc --disable_viewer --data_factor $DATA_FACTOR \
CUDA_VISIBLE_DEVICES=0 python examples/simple_trainer.py mcmc --disable_viewer --data_factor $DATA_FACTOR \
--strategy.cap-max $CAP_MAX \
--opacity_reg 0.001 \
--use_bilateral_grid \
Expand Down
4 changes: 2 additions & 2 deletions examples/benchmarks/mcmc.sh
Original file line number Diff line number Diff line change
Expand Up @@ -16,7 +16,7 @@ do
echo "Running $SCENE"

# train without eval
CUDA_VISIBLE_DEVICES=0 python simple_trainer.py mcmc --eval_steps -1 --disable_viewer --data_factor $DATA_FACTOR \
CUDA_VISIBLE_DEVICES=0 python examples/simple_trainer.py mcmc --eval_steps -1 --disable_viewer --data_factor $DATA_FACTOR \
--strategy.cap-max $CAP_MAX \
--render_traj_path $RENDER_TRAJ_PATH \
--data_dir $SCENE_DIR/$SCENE/ \
Expand All @@ -25,7 +25,7 @@ do
# run eval and render
for CKPT in $RESULT_DIR/$SCENE/ckpts/*;
do
CUDA_VISIBLE_DEVICES=0 python simple_trainer.py mcmc --disable_viewer --data_factor $DATA_FACTOR \
CUDA_VISIBLE_DEVICES=0 python examples/simple_trainer.py mcmc --disable_viewer --data_factor $DATA_FACTOR \
--strategy.cap-max $CAP_MAX \
--render_traj_path $RENDER_TRAJ_PATH \
--data_dir $SCENE_DIR/$SCENE/ \
Expand Down

0 comments on commit 5569018

Please sign in to comment.