Skip to content

Commit

Permalink
add missing train_cli.sh (fix #185) (#193)
Browse files Browse the repository at this point in the history
  • Loading branch information
WhenMelancholy authored May 28, 2024
1 parent 343f14b commit ac154fe
Showing 1 changed file with 28 additions and 0 deletions.
Original file line number Diff line number Diff line change
@@ -0,0 +1,28 @@
fairseq-train --user-dir ./graphormer --ddp-backend legacy_ddp --task graph_diffusion \
--criterion difussion_loss --arch graphormer_diff_base \
--data-path ./dataset/16sys_db \
--num-workers 1 \
--optimizer adam --adam-betas '(0.9, 0.98)' \
--lr 0 --weight-decay 1e-3 \
--clip-norm 10.0 \
--lr-scheduler polynomial_decay \
--max-epoch 1 \
--warmup-updates 1000 --total-num-update 10000 \
--batch-size 16 \
--train-subset 'placeholder' \
--valid-subset 'test_md' \
--validate-interval 1 \
--save-interval 5 \
--num-diffusion-timesteps 500 \
--diffusion-sampling 'ddpm' \
--ddim-steps 500 \
--diffusion-beta-schedule 'sigmoid' \
--diffusion-layer-add-time-emb \
--diffusion-layer-proj-time-emb \
--diffusion-beta-end 0.02 \
--ema-decay 0.9999 --store-ema --uses-ema --ema-fp32 \
--prior-distribution-std 1 \
--pairwise-loss \
--num-epsilon-estimator 1 \
--test-mode \
--finetune-from-model saved_checkpoints/checkpoint_prolig.pt

0 comments on commit ac154fe

Please sign in to comment.