-
Notifications
You must be signed in to change notification settings - Fork 0
/
exp.sh
executable file
·39 lines (34 loc) · 1.27 KB
/
exp.sh
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
36
37
38
39
# OMP_NUM_THREADS=1 python -m torch.distributed.launch --nproc_per_node=8 main.py \
# esm2_t12_35M_UR50D \
# /root/esm/examples/data/some_proteins.fasta output/ \
# --include contacts
OMP_NUM_THREADS=1 python -m torch.distributed.launch --nproc_per_node=2 main.py \
/junde/snRNA.fasta snRNA_35M_100epoch/ \
--include contacts \
--num_layers 12 \
--embed_dim 480 \
--coden_size 3
--fsdp
# OMP_NUM_THREADS=1 python -m torch.distributed.launch --nproc_per_node=2 main.py \
# /junde/snRNA.fasta snRNA_150M_100epoch/ \
# --include contacts \
# --num_layers 30 \
# --embed_dim 640
# OMP_NUM_THREADS=1 python -m torch.distributed.launch --nproc_per_node=2 main.py \
# /junde/snRNA.fasta snRNA_35M_coden1_100epoch/ \
# --include contacts \
# --num_layers 12 \
# --embed_dim 480 \
# --coden_size 1
# OMP_NUM_THREADS=1 python -m torch.distributed.launch --nproc_per_node=2 main.py \
# /junde/snRNA.fasta test/ \
# --include contacts \
# --num_layers 12 \
# --embed_dim 480 \
# --coden_size 1
# OMP_NUM_THREADS=1 python -m torch.distributed.launch --nproc_per_node=2 main.py \
# /junde/miRNA.fasta miRNA_35M_epoch100/ \
# --include contacts \
# --num_layers 12 \
# --embed_dim 480 \
# --coden_size 3