From 58cdcc2b41311b1bb632e9f9e4a7ad6c3ded6276 Mon Sep 17 00:00:00 2001
From: evonneng <evonne_ng@berkeley.edu>
Date: Wed, 11 May 2022 18:27:42 +0000
Subject: [PATCH] update model saving

---
 scripts/benchmarking/launch_train_blender.sh | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/scripts/benchmarking/launch_train_blender.sh b/scripts/benchmarking/launch_train_blender.sh
index 31165ea4c..2e96ab835 100755
--- a/scripts/benchmarking/launch_train_blender.sh
+++ b/scripts/benchmarking/launch_train_blender.sh
@@ -7,6 +7,8 @@ for dataset in ${DATASETS[@]}; do
     python scripts/run_train.py \
            data.dataset.data_directory=data/blender/${dataset} \
            experiment_name=blender_${dataset} \
+           graph.model_dir=mattport_models/ \
+           graph.steps_per_save=5000 \
            logging.enable_stats=False \
            logging.enable_profiler=False &
     echo "Launched ${dataset} on gpu ${!GPU_IDX}"