File size: 1,152 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
28
29
30
31
32
33
34
35
#! /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}