Skip to content

Commit 015fa97

Browse files
seq_emb.sh
1 parent 8278d40 commit 015fa97

1 file changed

Lines changed: 6 additions & 9 deletions

File tree

scripts/RNAERNIE/seq_emb.sh

Lines changed: 6 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -1,20 +1,17 @@
11
#!/bin/bash
22

3-
export model_location="/home/ma_run_ze/lzm/rnagym/fitness/baselines/RNA_FM/RNA-FM"
4-
export reference_sheet="/home/ma_run_ze/lzm/rnagym/fitness/reference_sheet_final.csv"
5-
export output_dir="/data_share/marunze/lzm/rnagym/fitness/embeddings/RNAernie/"
6-
export dms_data_dir="/home/ma_run_ze/lzm/rnagym/fitness/fitness_processed_assays"
3+
4+
export model_location=""
5+
export reference_sheet=""
6+
export output_dir=""
7+
export dms_data_dir=""
78
conda activate rnaernie
89
export model_checkpoint="src/"
910
export vocab_path="src/vocab_1MER.txt"
1011
mkdir -p "$output_dir"
11-
export HF_ENDPOINT=https://hf-mirror.com
12-
export CUDA_VISIBLE_DEVICES=4 # Set to the GPU you want to use
13-
# Get the current index from the array (0-31)
14-
# Run the scoring script with the array task ID
1512
python3 seq_emb.py \
1613
--output_dir_path "$output_dir" \
1714
--ref_sheet "$reference_sheet" \
1815
--dms_dir_path "$dms_data_dir" \
1916
--model_checkpoint "$model_checkpoint" \
20-
--vocab_path "$vocab_path"
17+
--vocab_path "$vocab_path"

0 commit comments

Comments
 (0)