File tree Expand file tree Collapse file tree 1 file changed +28
-0
lines changed
distributional_graphormer/protein-ligand/src/examples/diffusion Expand file tree Collapse file tree 1 file changed +28
-0
lines changed Original file line number Diff line number Diff line change 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
You can’t perform that action at this time.
0 commit comments