#!/bin/bash

# network
source /baai-cwm-1/baai_cwm_ml/algorithm/xiuyu.yang/clash.sh
bash /baai-cwm-1/baai_cwm_ml/algorithm/xiuyu.yang/clash-for-linux-backup/start.sh
proxy_on

# env
source /baai-cwm-1/baai_cwm_ml/algorithm/xiuyu.yang/anaconda3/etc/profile.d/conda.sh
conda config --append envs_dirs /baai-cwm-1/baai_cwm_ml/algorithm/xiuyu.yang/.conda/envs
conda activate traj

cd /baai-cwm-1/baai_cwm_ml/algorithm/xiuyu.yang/work/dev4/

export TORCH_LOGS='0'
export TF_CPP_MIN_LOG_LEVEL='2'
export TF_ENABLE_ONEDNN_OPTS='0'
export PYTHONPATH='.'

NUM_WORKERS=$1
SIM_DIR=$2

echo 'Start running ...'
python dev/metrics/compute_metrics.py --compute_metric --num_workers "$NUM_WORKERS" --sim_dir "$SIM_DIR" ${@:3}

echo 'Done!