Use custom variables in config for k-fold

[thapaliya19]

I am trying to run k-fold using a config. To do so, I have written a shell script that loops over the number of folds and runs the training script. I am using a single config file, where I have set a new variable, fold=1, and using the command line.

config_file=$1

# check if the config file exists

if [ ! -f $config_file ]; then
    echo "Config file does not exist"
    exit 1
fi

# run a loop to run training for the five folds
k=5
for i in $(seq 2 $k); do
    python mmdetection/tools/train.py $config_file --gpu-id 1 --cfg-options fold=$i
done

Inside the config file, I am using the fold variable to change the data directories to load the annotation files, e.g.:

ann_file=os.path.join(
                data_root, f"fold_{fold}", "annos_train.json"
            )

. But, when running the script, though the fold variable changes the values, the ann_file does not change. How can I accomplish this task?