diff options
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)" |