Skip to content

Commit ac154fe

Browse files
add missing train_cli.sh (fix #185) (#193)
1 parent 343f14b commit ac154fe

File tree

1 file changed

+28
-0
lines changed
  • distributional_graphormer/protein-ligand/src/examples/diffusion

1 file changed

+28
-0
lines changed
Lines changed: 28 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,28 @@
1+
fairseq-train --user-dir ./graphormer --ddp-backend legacy_ddp --task graph_diffusion \
2+
--criterion difussion_loss --arch graphormer_diff_base \
3+
--data-path ./dataset/16sys_db \
4+
--num-workers 1 \
5+
--optimizer adam --adam-betas '(0.9, 0.98)' \
6+
--lr 0 --weight-decay 1e-3 \
7+
--clip-norm 10.0 \
8+
--lr-scheduler polynomial_decay \
9+
--max-epoch 1 \
10+
--warmup-updates 1000 --total-num-update 10000 \
11+
--batch-size 16 \
12+
--train-subset 'placeholder' \
13+
--valid-subset 'test_md' \
14+
--validate-interval 1 \
15+
--save-interval 5 \
16+
--num-diffusion-timesteps 500 \
17+
--diffusion-sampling 'ddpm' \
18+
--ddim-steps 500 \
19+
--diffusion-beta-schedule 'sigmoid' \
20+
--diffusion-layer-add-time-emb \
21+
--diffusion-layer-proj-time-emb \
22+
--diffusion-beta-end 0.02 \
23+
--ema-decay 0.9999 --store-ema --uses-ema --ema-fp32 \
24+
--prior-distribution-std 1 \
25+
--pairwise-loss \
26+
--num-epsilon-estimator 1 \
27+
--test-mode \
28+
--finetune-from-model saved_checkpoints/checkpoint_prolig.pt

0 commit comments

Comments
 (0)