inference/scripts/infer_instanceseg.bash

exec_path="$1"

if [ -n "$2" ]; then
  rank_to_worldsize=$2-$(nvidia-smi -L | wc -l)
  echo running on rank $2, input will be partitioned by $rank_to_worldsize
  export CUDA_VISIBLE_DEVICES=$2
else
  rank_to_worldsize='-'
fi

echo $rank_to_worldsize

echo running instance segmentation for $exec_path
conda run --no-capture-output -n see_through_dev python inference/scripts/parse_live2d.py instance_segmentation --exec_list $exec_path --rank_to_worldsize $rank_to_worldsize