File size: 724 Bytes
c1a7f73
 
d37e5d1
 
 
 
 
 
 
 
 
 
 
 
c1a7f73
 
d37e5d1
c1a7f73
 
 
 
 
 
 
 
d37e5d1
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
#!/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!