updated benchmarking to match new config

This commit is contained in:
evonneng committed 2022-06-14 00:15:05 +00:00
1 parent 3921839d1a
commit b2648c6522
1 file changed
+2 -2
+2 -2
View File
@@ -40,8 +40,8 @@ for dataset in ${DATASETS[@]}; do
python scripts/run_train.py \
--config-name ${METHOD} \
'~logging.writer.LocalWriter' \
data.dataset.data_directory=data/blender/${dataset} \
data.dataset.downscale_factor=1 \
data.dataset_inputs_train.data_directory=data/blender/${dataset} \
data.dataset_inputs_eval.data_directory=data/blender/${dataset} \
experiment_name=blender_${dataset}_${tag} \
graph.model_dir=pyrad_models/ \
graph.steps_per_save=25000 \