test.sh

export CUDA_VISIBLE_DEVICES=0
accelerate launch test.py --config config/run_two_man.yaml