This commit is contained in:
Mufan Qiu 2024-05-28 14:22:34 +08:00 коммит произвёл GitHub
Родитель 343f14b283
Коммит ac154fe425
Не найден ключ, соответствующий данной подписи
Идентификатор ключа GPG: B5690EEEBB952194
1 изменённых файлов: 28 добавлений и 0 удалений

Просмотреть файл

@ -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