diff options
| author | YurenHao0426 <Blackhao0426@gmail.com> | 2026-06-29 12:04:47 -0500 |
|---|---|---|
| committer | YurenHao0426 <Blackhao0426@gmail.com> | 2026-06-29 12:04:47 -0500 |
| commit | c54ddb88b532be28ca3096e21de405d90163ecfa (patch) | |
| tree | 3270ec9269dbee14ea915963f0d28e933303d5a7 /diag/run_pna.sh | |
| parent | d12722525fc010a3910b5152c72654a2ade5eac4 (diff) | |
Package full RRoG GNN project
Diffstat (limited to 'diag/run_pna.sh')
| -rw-r--r-- | diag/run_pna.sh | 3 |
1 files changed, 1 insertions, 2 deletions
diff --git a/diag/run_pna.sh b/diag/run_pna.sh index 897315d..157d8ac 100644 --- a/diag/run_pna.sh +++ b/diag/run_pna.sh @@ -4,10 +4,9 @@ cd /home/yurenh2/rrog export PYTHONPATH=/home/yurenh2/rrog echo "PNA start gpu=${CUDA_VISIBLE_DEVICES:-?} $(date -Is)" for s in 0 1 2 3 4; do - ck=runs/ckpt_color_pna_full_none_n50_k3_p0.2_T3_ns3_s${s}.pt + ck=runs/ckpt_color_rrog_trm_pna_full_none_n50_k3_p0.2_T3_ns3_s${s}.pt echo "== pna s$s ==" python3 diag/train_color.py --mode train --conv pna --pe none --p 0.2 --epochs 150 --seed "$s" || echo "!! train pna s$s" python3 diag/train_color.py --mode le --ckpt "$ck" || echo "!! le pna s$s" - python3 diag/ptrm_color.py --ckpt "$ck" --K 16 --n_graphs 150 --sigmas 0.2 || echo "!! ptrm pna s$s" done echo "donePNA=$(date -Is)" |