#! /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/

# env
export REQUESTS_CA_BUNDLE="/etc/ssl/certs/ca-certificates.crt"
export HTTPS_PROXY="https://192.168.0.10:443/"
export https_proxy="https://192.168.0.10:443/"

export WANDB=1

# args
DEVICES=$1
CONFIG='configs/ours_long_term.yaml'
# CKPT_PATH='output/scalable_smart_long/last.ckpt'
# CKPT_PATH='output2/seq_10_150_3_3_encode_occ_separate_offsets_bs8_128_no_seqindex_long/last.ckpt'
CKPT_PATH='output2/seq_10_150_3_3_encode_occ_separate_offsets_bs8_128_no_seqindex_long_lastvalid_ablation_grid/last.ckpt'

# run
PYTHONPATH=".":$PYTHONPATH python3 run.py \
                            --devices $DEVICES \
                            --config $CONFIG \
                            --ckpt_path $CKPT_PATH ${@:2}