From 926742128dede2240fbff6c46a068b17686b1be9 Mon Sep 17 00:00:00 2001 From: Friedrich Branschke Date: Wed, 4 Dec 2024 14:53:44 +0100 Subject: [PATCH] compression benchmarks use full resolution for training --- examples/benchmarks/compression/mcmc.sh | 10 ++-------- 1 file changed, 2 insertions(+), 8 deletions(-) diff --git a/examples/benchmarks/compression/mcmc.sh b/examples/benchmarks/compression/mcmc.sh index b5df5058..bcc1fcfb 100644 --- a/examples/benchmarks/compression/mcmc.sh +++ b/examples/benchmarks/compression/mcmc.sh @@ -21,22 +21,16 @@ CAP_MAX=1000000 for SCENE in $SCENE_LIST; do - if [ "$SCENE" = "bonsai" ] || [ "$SCENE" = "counter" ] || [ "$SCENE" = "kitchen" ] || [ "$SCENE" = "room" ]; then - DATA_FACTOR=2 - else - DATA_FACTOR=4 - fi - 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 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 $DATA_FACTOR \ + CUDA_VISIBLE_DEVICES=0 python simple_trainer.py mcmc --disable_viewer --data_factor 1 \ --strategy.cap-max $CAP_MAX \ --data_dir $SCENE_DIR/$SCENE/ \ --result_dir $RESULT_DIR/$SCENE/ \