#!/bin/bash

# A batch script for running a job on Oscar's 3090 condo, using the Slurm scheduler
# The 3090 condo runs NVIDIA's GeForce RTX 3090 graphics card

#SBATCH -p 3090-gcondo --gres=gpu:1
#SBATCH --constraint=a6000|l40s|geforce3090
#SBATCH --exclude=gpu1506,gpu2108,gpu2109,gpu2112,gpu2113,gpu2114,gpu2115,gpu2116
#SBATCH -N 1 # gives one node, makes sure cpu cores are on same node
#SBATCH -c 1 # num CPU cores
#SBATCH --mem=24G
#SBATCH -t 2:00:00
#SBATCH -e output/slurm_logs/%j.err
#SBATCH -o output/slurm_logs/%j.out
#SBATCH --mail-user=nate_gillman@brown.edu
#SBATCH --mail-type=ALL

# SET UP COMPUTING ENV
export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/lib/nvidia

# Activate virtual environment
# Load anaconda module, and other modules
source /gpfs/runtime/opt/anaconda/2020.02/etc/profile.d/conda.sh

export CUDA_VISIBLE_DEVICES=0 # e.g. 0 or 0,1,2,3
module load cuda/12.2.0-4lgnkrh
CONDA_ENV_DIR=/oscar/data/superlab/users/nates_stuff/cogvideox-controlnet-clean/conda-env
conda activate $CONDA_ENV_DIR
HOME_DIR=/oscar/data/superlab/users/nates_stuff/cogvideox-controlnet-clean
cd ${HOME_DIR}

# python scripts/print_csv_paths.py datasets/point-force/test/benchmark/apple

####################################################################################
####################################################################################
####################################################################################
################# OUR FINAL BENCHMARK FOR POINT FORCE POKEING ######################
####################################################################################
####################################################################################
####################################################################################
IMAGE_CSVS=("datasets/point-force/test/benchmark/apple/_apple1_obj1_prompt1.csv" "datasets/point-force/test/benchmark/apple/_apple1_obj1_prompt2.csv" "datasets/point-force/test/benchmark/apple/_apple2_obj1_prompt1.csv" "datasets/point-force/test/benchmark/apple/_apple2_obj1_prompt2.csv" "datasets/point-force/test/benchmark/apple/_apple3_obj1_prompt1.csv" "datasets/point-force/test/benchmark/apple/_apple3_obj1_prompt2.csv" "datasets/point-force/test/benchmark/apple/_apple4_obj1_prompt1.csv" "datasets/point-force/test/benchmark/apple/_apple4_obj1_prompt2.csv") 
IMAGE_CSVS=("datasets/point-force/test/benchmark/balloon/_balloon3_obj1_prompt1.csv" "datasets/point-force/test/benchmark/balloon/_balloon3_obj1_prompt2.csv" "datasets/point-force/test/benchmark/balloon/_balloon4_obj1_prompt1.csv" "datasets/point-force/test/benchmark/balloon/_balloon4_obj1_prompt2.csv") 
IMAGE_CSVS=("datasets/point-force/test/benchmark/blueberrybush/_blueberrybush1_obj1_prompt1.csv" "datasets/point-force/test/benchmark/blueberrybush/_blueberrybush1_obj1_prompt2.csv" "datasets/point-force/test/benchmark/blueberrybush/_blueberrybush2_obj1_prompt1.csv" "datasets/point-force/test/benchmark/blueberrybush/_blueberrybush2_obj1_prompt2.csv" "datasets/point-force/test/benchmark/blueberrybush/_blueberrybush3_obj1_prompt1.csv" "datasets/point-force/test/benchmark/blueberrybush/_blueberrybush3_obj1_prompt2.csv" "datasets/point-force/test/benchmark/blueberrybush/_blueberrybush4_obj1_prompt1.csv" "datasets/point-force/test/benchmark/blueberrybush/_blueberrybush4_obj1_prompt2.csv") 
IMAGE_CSVS=("datasets/point-force/test/benchmark/dandelion/_dandelion1_obj1_prompt1.csv" "datasets/point-force/test/benchmark/dandelion/_dandelion1_obj1_prompt2.csv" "datasets/point-force/test/benchmark/dandelion/_dandelion3_obj1_prompt1.csv" "datasets/point-force/test/benchmark/dandelion/_dandelion3_obj1_prompt2.csv" "datasets/point-force/test/benchmark/dandelion/_dandelion4_obj1_prompt1.csv" "datasets/point-force/test/benchmark/dandelion/_dandelion4_obj1_prompt2.csv")
IMAGE_CSVS=("datasets/point-force/test/benchmark/ivy/_ivy1_obj1_prompt2.csv")
IMAGE_CSVS=("datasets/point-force/test/benchmark/ornament/_ornament1_obj1_prompt1.csv" "datasets/point-force/test/benchmark/ornament/_ornament1_obj1_prompt2.csv" "datasets/point-force/test/benchmark/ornament/_ornament2_obj1_prompt1.csv" "datasets/point-force/test/benchmark/ornament/_ornament2_obj1_prompt2.csv" "datasets/point-force/test/benchmark/ornament/_ornament4_obj1_prompt1.csv") 
IMAGE_CSVS=("datasets/point-force/test/benchmark/rose/_rose2_obj1_prompt1.csv" "datasets/point-force/test/benchmark/rose/_rose2_obj1_prompt2.csv" "datasets/point-force/test/benchmark/rose/_rose2_obj1_prompt3.csv" "datasets/point-force/test/benchmark/rose/_rose2_obj1_prompt4.csv" "datasets/point-force/test/benchmark/rose/_rose2_obj1_prompt5.csv" "datasets/point-force/test/benchmark/rose/_rose2_obj1_prompt6.csv" "datasets/point-force/test/benchmark/rose/_rose3_obj1_prompt1.csv" "datasets/point-force/test/benchmark/rose/_rose3_obj1_prompt2.csv" "datasets/point-force/test/benchmark/rose/_rose3_obj1_prompt3.csv" "datasets/point-force/test/benchmark/rose/_rose3_obj1_prompt4.csv" "datasets/point-force/test/benchmark/rose/_rose3_obj1_prompt5.csv" "datasets/point-force/test/benchmark/rose/_rose3_obj1_prompt6.csv" "datasets/point-force/test/benchmark/rose/_rose4_obj1_prompt1.csv" "datasets/point-force/test/benchmark/rose/_rose4_obj1_prompt2.csv" "datasets/point-force/test/benchmark/rose/_rose4_obj1_prompt3.csv" "datasets/point-force/test/benchmark/rose/_rose4_obj1_prompt4.csv" "datasets/point-force/test/benchmark/rose/_rose4_obj1_prompt5.csv" "datasets/point-force/test/benchmark/rose/_rose4_obj1_prompt6.csv" "datasets/point-force/test/benchmark/rose/_rose5_obj1_prompt1.csv" "datasets/point-force/test/benchmark/rose/_rose5_obj1_prompt2.csv" "datasets/point-force/test/benchmark/rose/_rose5_obj1_prompt3.csv" "datasets/point-force/test/benchmark/rose/_rose5_obj1_prompt4.csv" "datasets/point-force/test/benchmark/rose/_rose5_obj2_prompt1.csv" "datasets/point-force/test/benchmark/rose/_rose5_obj2_prompt2.csv" "datasets/point-force/test/benchmark/rose/_rose5_obj2_prompt3.csv" "datasets/point-force/test/benchmark/rose/_rose5_obj2_prompt4.csv") 
IMAGE_CSVS=("datasets/point-force/test/benchmark/sunflower/_sunflower2_obj1_prompt1.csv" "datasets/point-force/test/benchmark/sunflower/_sunflower2_obj1_prompt2.csv" "datasets/point-force/test/benchmark/sunflower/_sunflower3_obj1_prompt1.csv" "datasets/point-force/test/benchmark/sunflower/_sunflower3_obj1_prompt2.csv") 
IMAGE_CSVS=("datasets/point-force/test/benchmark/swing/_swing3_obj1_prompt1.csv" "datasets/point-force/test/benchmark/swing/_swing3_obj1_prompt2.csv") 
IMAGE_CSVS=("datasets/point-force/test/benchmark/toycar/_toycar1_obj1_prompt2.csv" "datasets/point-force/test/benchmark/toycar/_toycar2_obj1_prompt1_vary_speeds.csv" "datasets/point-force/test/benchmark/toycar/_toycar2_obj1_prompt2.csv" "datasets/point-force/test/benchmark/toycar/_toycar3_obj1_prompt1_vary_speeds.csv" "datasets/point-force/test/benchmark/toycar/_toycar3_obj1_prompt2.csv" "datasets/point-force/test/benchmark/toycar/_toycar4_obj1_prompt1_vary_speeds.csv" "datasets/point-force/test/benchmark/toycar/_toycar4_obj1_prompt2.csv")
IMAGE_CSVS=("datasets/point-force/test/benchmark/toytrainontracks/_toytrainontrack1_obj1_prompt1.csv" "datasets/point-force/test/benchmark/toytrainontracks/_toytrainontrack1_obj2_prompt1.csv" "datasets/point-force/test/benchmark/toytrainontracks/_toytrainontrack2_obj1_prompt1.csv" "datasets/point-force/test/benchmark/toytrainontracks/_toytrainontrack2_obj2_prompt1.csv" "datasets/point-force/test/benchmark/toytrainontracks/_toytrainontrack2_obj3_prompt1.csv" "datasets/point-force/test/benchmark/toytrainontracks/_toytrainontrack3_obj1_prompt1.csv" "datasets/point-force/test/benchmark/toytrainontracks/_toytrainontrack3_obj2_prompt1.csv" "datasets/point-force/test/benchmark/toytrainontracks/_toytrainontrack3_obj3_prompt1.csv" "datasets/point-force/test/benchmark/toytrainontracks/_toytrainontrack5_obj1_prompt1.csv" "datasets/point-force/test/benchmark/toytrainontracks/_toytrainontrack5_obj2_prompt1.csv" "datasets/point-force/test/benchmark/toytrainontracks/_toytrainontrack6_obj1_prompt1.csv" "datasets/point-force/test/benchmark/toytrainontracks/_toytrainontrack6_obj1_prompt2.csv" "datasets/point-force/test/benchmark/toytrainontracks/_toytrainontrack6_obj2_prompt1.csv" "datasets/point-force/test/benchmark/toytrainontracks/_toytrainontrack6_obj2_prompt2.csv" "datasets/point-force/test/benchmark/toytrainontracks/_toytrainontrack7_obj1_prompt1.csv" "datasets/point-force/test/benchmark/toytrainontracks/_toytrainontrack7_obj2_prompt1.csv" "datasets/point-force/test/benchmark/toytrainontracks/_toytrainontrack7_obj3_prompt1.csv")
IMAGE_CSVS=("datasets/point-force/test/benchmark/windmill/_windmill2_obj1_prompt1.csv" "datasets/point-force/test/benchmark/windmill/_windmill2_obj1_prompt2.csv")
# all of them concatenated
IMAGE_CSVS=("datasets/point-force/test/benchmark/apple/_apple1_obj1_prompt1.csv" "datasets/point-force/test/benchmark/apple/_apple1_obj1_prompt2.csv" "datasets/point-force/test/benchmark/apple/_apple2_obj1_prompt1.csv" "datasets/point-force/test/benchmark/apple/_apple2_obj1_prompt2.csv" "datasets/point-force/test/benchmark/apple/_apple3_obj1_prompt1.csv" "datasets/point-force/test/benchmark/apple/_apple3_obj1_prompt2.csv" "datasets/point-force/test/benchmark/apple/_apple4_obj1_prompt1.csv" "datasets/point-force/test/benchmark/apple/_apple4_obj1_prompt2.csv" "datasets/point-force/test/benchmark/balloon/_balloon3_obj1_prompt1.csv" "datasets/point-force/test/benchmark/balloon/_balloon3_obj1_prompt2.csv" "datasets/point-force/test/benchmark/balloon/_balloon4_obj1_prompt1.csv" "datasets/point-force/test/benchmark/balloon/_balloon4_obj1_prompt2.csv" "datasets/point-force/test/benchmark/blueberrybush/_blueberrybush1_obj1_prompt1.csv" "datasets/point-force/test/benchmark/blueberrybush/_blueberrybush1_obj1_prompt2.csv" "datasets/point-force/test/benchmark/blueberrybush/_blueberrybush2_obj1_prompt1.csv" "datasets/point-force/test/benchmark/blueberrybush/_blueberrybush2_obj1_prompt2.csv" "datasets/point-force/test/benchmark/blueberrybush/_blueberrybush3_obj1_prompt1.csv" "datasets/point-force/test/benchmark/blueberrybush/_blueberrybush3_obj1_prompt2.csv" "datasets/point-force/test/benchmark/blueberrybush/_blueberrybush4_obj1_prompt1.csv" "datasets/point-force/test/benchmark/blueberrybush/_blueberrybush4_obj1_prompt2.csv" "datasets/point-force/test/benchmark/dandelion/_dandelion1_obj1_prompt1.csv" "datasets/point-force/test/benchmark/dandelion/_dandelion1_obj1_prompt2.csv" "datasets/point-force/test/benchmark/dandelion/_dandelion3_obj1_prompt1.csv" "datasets/point-force/test/benchmark/dandelion/_dandelion3_obj1_prompt2.csv" "datasets/point-force/test/benchmark/dandelion/_dandelion4_obj1_prompt1.csv" "datasets/point-force/test/benchmark/dandelion/_dandelion4_obj1_prompt2.csv" "datasets/point-force/test/benchmark/ivy/_ivy1_obj1_prompt2.csv" "datasets/point-force/test/benchmark/ornament/_ornament1_obj1_prompt1.csv" "datasets/point-force/test/benchmark/ornament/_ornament1_obj1_prompt2.csv" "datasets/point-force/test/benchmark/ornament/_ornament2_obj1_prompt1.csv" "datasets/point-force/test/benchmark/ornament/_ornament2_obj1_prompt2.csv" "datasets/point-force/test/benchmark/ornament/_ornament4_obj1_prompt1.csv" "datasets/point-force/test/benchmark/rose/_rose2_obj1_prompt1.csv" "datasets/point-force/test/benchmark/rose/_rose2_obj1_prompt2.csv" "datasets/point-force/test/benchmark/rose/_rose2_obj1_prompt3.csv" "datasets/point-force/test/benchmark/rose/_rose2_obj1_prompt4.csv" "datasets/point-force/test/benchmark/rose/_rose2_obj1_prompt5.csv" "datasets/point-force/test/benchmark/rose/_rose2_obj1_prompt6.csv" "datasets/point-force/test/benchmark/rose/_rose3_obj1_prompt1.csv" "datasets/point-force/test/benchmark/rose/_rose3_obj1_prompt2.csv" "datasets/point-force/test/benchmark/rose/_rose3_obj1_prompt3.csv" "datasets/point-force/test/benchmark/rose/_rose3_obj1_prompt4.csv" "datasets/point-force/test/benchmark/rose/_rose3_obj1_prompt5.csv" "datasets/point-force/test/benchmark/rose/_rose3_obj1_prompt6.csv" "datasets/point-force/test/benchmark/rose/_rose4_obj1_prompt1.csv" "datasets/point-force/test/benchmark/rose/_rose4_obj1_prompt2.csv" "datasets/point-force/test/benchmark/rose/_rose4_obj1_prompt3.csv" "datasets/point-force/test/benchmark/rose/_rose4_obj1_prompt4.csv" "datasets/point-force/test/benchmark/rose/_rose4_obj1_prompt5.csv" "datasets/point-force/test/benchmark/rose/_rose4_obj1_prompt6.csv" "datasets/point-force/test/benchmark/rose/_rose5_obj1_prompt1.csv" "datasets/point-force/test/benchmark/rose/_rose5_obj1_prompt2.csv" "datasets/point-force/test/benchmark/rose/_rose5_obj1_prompt3.csv" "datasets/point-force/test/benchmark/rose/_rose5_obj1_prompt4.csv" "datasets/point-force/test/benchmark/rose/_rose5_obj2_prompt1.csv" "datasets/point-force/test/benchmark/rose/_rose5_obj2_prompt2.csv" "datasets/point-force/test/benchmark/rose/_rose5_obj2_prompt3.csv" "datasets/point-force/test/benchmark/rose/_rose5_obj2_prompt4.csv" "datasets/point-force/test/benchmark/sunflower/_sunflower2_obj1_prompt1.csv" "datasets/point-force/test/benchmark/sunflower/_sunflower2_obj1_prompt2.csv" "datasets/point-force/test/benchmark/sunflower/_sunflower3_obj1_prompt1.csv" "datasets/point-force/test/benchmark/sunflower/_sunflower3_obj1_prompt2.csv" "datasets/point-force/test/benchmark/swing/_swing3_obj1_prompt1.csv" "datasets/point-force/test/benchmark/swing/_swing3_obj1_prompt2.csv" "datasets/point-force/test/benchmark/toycar/_toycar1_obj1_prompt2.csv" "datasets/point-force/test/benchmark/toycar/_toycar2_obj1_prompt1_vary_speeds.csv" "datasets/point-force/test/benchmark/toycar/_toycar2_obj1_prompt2.csv" "datasets/point-force/test/benchmark/toycar/_toycar3_obj1_prompt1_vary_speeds.csv" "datasets/point-force/test/benchmark/toycar/_toycar3_obj1_prompt2.csv" "datasets/point-force/test/benchmark/toycar/_toycar4_obj1_prompt1_vary_speeds.csv" "datasets/point-force/test/benchmark/toycar/_toycar4_obj1_prompt2.csv" "datasets/point-force/test/benchmark/toytrainontracks/_toytrainontrack1_obj1_prompt1.csv" "datasets/point-force/test/benchmark/toytrainontracks/_toytrainontrack1_obj2_prompt1.csv" "datasets/point-force/test/benchmark/toytrainontracks/_toytrainontrack2_obj1_prompt1.csv" "datasets/point-force/test/benchmark/toytrainontracks/_toytrainontrack2_obj2_prompt1.csv" "datasets/point-force/test/benchmark/toytrainontracks/_toytrainontrack2_obj3_prompt1.csv" "datasets/point-force/test/benchmark/toytrainontracks/_toytrainontrack3_obj1_prompt1.csv" "datasets/point-force/test/benchmark/toytrainontracks/_toytrainontrack3_obj2_prompt1.csv" "datasets/point-force/test/benchmark/toytrainontracks/_toytrainontrack3_obj3_prompt1.csv" "datasets/point-force/test/benchmark/toytrainontracks/_toytrainontrack5_obj1_prompt1.csv" "datasets/point-force/test/benchmark/toytrainontracks/_toytrainontrack5_obj2_prompt1.csv" "datasets/point-force/test/benchmark/toytrainontracks/_toytrainontrack6_obj1_prompt1.csv" "datasets/point-force/test/benchmark/toytrainontracks/_toytrainontrack6_obj1_prompt2.csv" "datasets/point-force/test/benchmark/toytrainontracks/_toytrainontrack6_obj2_prompt1.csv" "datasets/point-force/test/benchmark/toytrainontracks/_toytrainontrack6_obj2_prompt2.csv" "datasets/point-force/test/benchmark/toytrainontracks/_toytrainontrack7_obj1_prompt1.csv" "datasets/point-force/test/benchmark/toytrainontracks/_toytrainontrack7_obj2_prompt1.csv" "datasets/point-force/test/benchmark/toytrainontracks/_toytrainontrack7_obj3_prompt1.csv" "datasets/point-force/test/benchmark/windmill/_windmill2_obj1_prompt1.csv" "datasets/point-force/test/benchmark/windmill/_windmill2_obj1_prompt2.csv")


# Benchmark for mass_understanding_quantitative
# python scripts/print_csv_paths_all.py datasets/point-force/test/mass_understanding_quantitative
IMAGE_CSVS=("datasets/point-force/test/mass_understanding_quantitative/dirt/_materialballrollingballondirtbowling1_obj1_prompt1.csv" "datasets/point-force/test/mass_understanding_quantitative/dirt/_materialballrollingballondirtbowling2_obj1_prompt1.csv" "datasets/point-force/test/mass_understanding_quantitative/dirt/_materialballrollingballondirtbowling3_obj1_prompt1.csv" "datasets/point-force/test/mass_understanding_quantitative/dirt/_materialballrollingballondirtsoccer1_obj1_prompt1.csv" "datasets/point-force/test/mass_understanding_quantitative/dirt/_materialballrollingballondirtsoccer2_obj1_prompt1.csv" "datasets/point-force/test/mass_understanding_quantitative/dirt/_materialballrollingballondirtsoccer3_obj1_prompt1.csv" "datasets/point-force/test/mass_understanding_quantitative/grass/_materialballrollingballongrassbowling1_obj1_prompt1.csv" "datasets/point-force/test/mass_understanding_quantitative/grass/_materialballrollingballongrassbowling2_obj1_prompt1.csv" "datasets/point-force/test/mass_understanding_quantitative/grass/_materialballrollingballongrassbowling3_obj1_prompt1.csv" "datasets/point-force/test/mass_understanding_quantitative/grass/_materialballrollingballongrasssoccer1_obj1_prompt1.csv" "datasets/point-force/test/mass_understanding_quantitative/grass/_materialballrollingballongrasssoccer2_obj1_prompt1.csv" "datasets/point-force/test/mass_understanding_quantitative/grass/_materialballrollingballongrasssoccer3_obj1_prompt1.csv" "datasets/point-force/test/mass_understanding_quantitative/stone/_materialballrollingballonstonebowling1_obj1_prompt1.csv" "datasets/point-force/test/mass_understanding_quantitative/stone/_materialballrollingballonstonebowling2_obj1_prompt1.csv" "datasets/point-force/test/mass_understanding_quantitative/stone/_materialballrollingballonstonebowling3_obj1_prompt1.csv" "datasets/point-force/test/mass_understanding_quantitative/stone/_materialballrollingballonstonesoccer1_obj1_prompt1.csv" "datasets/point-force/test/mass_understanding_quantitative/stone/_materialballrollingballonstonesoccer2_obj1_prompt1.csv" "datasets/point-force/test/mass_understanding_quantitative/stone/_materialballrollingballonstonesoccer3_obj1_prompt1.csv" "datasets/point-force/test/mass_understanding_quantitative/wood/_materialballrollingballonwoodbowling1_obj1_prompt1.csv" "datasets/point-force/test/mass_understanding_quantitative/wood/_materialballrollingballonwoodbowling2_obj1_prompt1.csv" "datasets/point-force/test/mass_understanding_quantitative/wood/_materialballrollingballonwoodbowling3_obj1_prompt1.csv" "datasets/point-force/test/mass_understanding_quantitative/wood/_materialballrollingballonwoodsoccer1_obj1_prompt1.csv" "datasets/point-force/test/mass_understanding_quantitative/wood/_materialballrollingballonwoodsoccer2_obj1_prompt1.csv" "datasets/point-force/test/mass_understanding_quantitative/wood/_materialballrollingballonwoodsoccer3_obj1_prompt1.csv")

# Benchmark for mass_understanding_qualitative
# python scripts/print_csv_paths_all.py datasets/point-force/test/mass_understanding_quantitative
IMAGE_CSVS=("datasets/point-force/test/mass_understanding_qualitative/cube/_materialexpcube1_obj1_prompt1.csv" "datasets/point-force/test/mass_understanding_qualitative/cube/_materialexpcube2_obj1_prompt1.csv" "datasets/point-force/test/mass_understanding_qualitative/cube/_materialexpcube3_obj1_prompt1.csv" "datasets/point-force/test/mass_understanding_qualitative/cube/_materialexpcube3_obj2_prompt1.csv" "datasets/point-force/test/mass_understanding_qualitative/laundrybasket/_materialexplaundrybasket1_obj1_prompt1.csv" "datasets/point-force/test/mass_understanding_qualitative/laundrybasket/_materialexplaundrybasket2_obj1_prompt1.csv" "datasets/point-force/test/mass_understanding_qualitative/ornament/_materialornament4_obj1_prompt1.csv" "datasets/point-force/test/mass_understanding_qualitative/ornament/_materialornament5_obj1_prompt1.csv")
IMAGE_CSVS=("datasets/point-force/test/mass_understanding_qualitative/skateboard/_materialexpskateboard1_obj1_prompt1.csv" "datasets/point-force/test/mass_understanding_qualitative/skateboard/_materialexpskateboard2_obj1_prompt1.csv" "datasets/point-force/test/mass_understanding_qualitative/smallcube/_materialexpsmallcube1_obj1_prompt1.csv" "datasets/point-force/test/mass_understanding_qualitative/smallcube/_materialexpsmallcube2_obj1_prompt1.csv" "datasets/point-force/test/mass_understanding_qualitative/smallcube/_materialexpsmallcube3_obj1_prompt1.csv" "datasets/point-force/test/mass_understanding_qualitative/smallcube/_materialexpsmallcube4_obj1_prompt1.csv" "datasets/point-force/test/mass_understanding_qualitative/smallcube/_materialexpsmallcube4_obj2_prompt1.csv" "datasets/point-force/test/mass_understanding_qualitative/solocup/_materialexpsolocup1_obj1_prompt1.csv") 
IMAGE_CSVS=("datasets/point-force/test/mass_understanding_qualitative/solocup/_materialexpsolocup2_obj1_prompt1.csv" "datasets/point-force/test/mass_understanding_qualitative/stackofbooks/_materialexpstackofbooks1_obj1_prompt1.csv" "datasets/point-force/test/mass_understanding_qualitative/stackofbooks/_materialexpstackofbooks2_obj1_prompt1.csv" "datasets/point-force/test/mass_understanding_qualitative/stackofbooks/_materialexpstackofbooks3_obj1_prompt1.csv" "datasets/point-force/test/mass_understanding_qualitative/stackofbooks/_materialexpstackofbooks3_obj2_prompt1.csv" "datasets/point-force/test/mass_understanding_qualitative/stackofbooks/_materialexpstackofbooks4_obj1_prompt1.csv" "datasets/point-force/test/mass_understanding_qualitative/stackofbooks/_materialexpstackofbooks4_obj2_prompt1.csv")



#######################################################
#######################################################
#######################################################
##### inference using a controlnet-trained model ######
#######################################################
#######################################################
#######################################################
# CHECKPOINTS=".cache/step-5000-checkpoint-point-force.pt"
CHECKPOINTS="output/point_force/2025-05-08_17-27-30/step-5000-checkpoint.pt"
MODEL_TYPE="controlnet_with_force_control_signal"

for ckpt in "${CHECKPOINTS[@]}"; do
    for image_csv in "${IMAGE_CSVS[@]}"; do
        bash scripts/point_force_inference_1_gpu.sh \
            --model_type "${MODEL_TYPE}" \
            --num_validation_videos 4 \
            --csv_path_val "${image_csv}" \
            --pretrained_controlnet_path "${ckpt}"
    done
done