-
Notifications
You must be signed in to change notification settings - Fork 2
/
Copy pathrun_AMP.sh
12 lines (12 loc) · 1.83 KB
/
run_AMP.sh
1
2
3
4
5
6
7
8
9
10
11
12
EXPERIMENT='EXPERIMENT_NAME'
BS=112
DEVICE=0
EPOCHS=300
python main.py --device $DEVICE --use_gpu --conv_encode_edge --block res+ --gcn_aggr softmax --learn_t --t 1.0 --dropout 0.2 --cross_val 1 --batch_size $BS --binary --save $EXPERIMENT --balanced_loader --metadata --num_metadata 8 --epochs $EPOCHS
python main.py --device $DEVICE --use_gpu --conv_encode_edge --block res+ --gcn_aggr softmax --learn_t --t 1.0 --dropout 0.2 --cross_val 2 --batch_size $BS --binary --save $EXPERIMENT --balanced_loader --metadata --num_metadata 8 --epochs $EPOCHS
python main.py --device $DEVICE --use_gpu --conv_encode_edge --block res+ --gcn_aggr softmax --learn_t --t 1.0 --dropout 0.2 --cross_val 3 --batch_size $BS --binary --save $EXPERIMENT --balanced_loader --metadata --num_metadata 8 --epochs $EPOCHS
python main.py --device $DEVICE --use_gpu --conv_encode_edge --block res+ --gcn_aggr softmax --learn_t --t 1.0 --dropout 0.2 --cross_val 4 --batch_size $BS --binary --save $EXPERIMENT --balanced_loader --metadata --num_metadata 8 --epochs $EPOCHS
python eval_validation.py --device $DEVICE --use_gpu --conv_encode_edge --block res+ --gcn_aggr softmax --learn_t --t 1.0 --dropout 0.2 --batch_size $BS --binary --save $EXPERIMENT --metadata --num_metadata 8
python eval_train.py --device $DEVICE --use_gpu --conv_encode_edge --block res+ --gcn_aggr softmax --learn_t --t 1.0 --dropout 0.2 --batch_size $BS --binary --save $EXPERIMENT --metadata --num_metadata 8
python eval_test.py --device $DEVICE --use_gpu --conv_encode_edge --block res+ --gcn_aggr softmax --learn_t --t 1.0 --dropout 0.2 --batch_size $BS --binary --save $EXPERIMENT --metadata --num_metadata 8
python test_ensamble.py --device $DEVICE --use_gpu --conv_encode_edge --block res+ --gcn_aggr softmax --learn_t --t 1.0 --dropout 0.2 --batch_size $BS --binary --save $EXPERIMENT --metadata --num_metadata 8