From a15185ea47ea6599b120e3682b81655a49dfb8d9 Mon Sep 17 00:00:00 2001 From: evonneng Date: Wed, 11 May 2022 19:22:20 +0000 Subject: [PATCH] fixed train iteration to much longer --- scripts/benchmarking/launch_train_blender.sh | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/scripts/benchmarking/launch_train_blender.sh b/scripts/benchmarking/launch_train_blender.sh index 2e96ab835..803a0be76 100755 --- a/scripts/benchmarking/launch_train_blender.sh +++ b/scripts/benchmarking/launch_train_blender.sh @@ -8,7 +8,8 @@ for dataset in ${DATASETS[@]}; do data.dataset.data_directory=data/blender/${dataset} \ experiment_name=blender_${dataset} \ graph.model_dir=mattport_models/ \ - graph.steps_per_save=5000 \ + graph.steps_per_save=25000 \ + graph.max_num_iterations=2000000 \ logging.enable_stats=False \ logging.enable_profiler=False & echo "Launched ${dataset} on gpu ${!GPU_IDX}"