Remove inference_tox21.py from the gpu example test
@@ -23,18 +23,12 @@ do
# Tox21 classification task with only one label
out_dir=nr_ar_${method}
python train_tox21.py --method ${method} --label NR-AR --conv-layers 1 --gpu ${gpu} --epoch 1 --unit-num 10 --out ${out_dir} --batchsize 32 --num-data=${tox21_num_data}
- python inference_tox21.py --in-dir ${out_dir} --gpu ${gpu} --num-data=${tox21_num_data}
python predict_tox21_with_classifier.py --in-dir ${out_dir} --gpu ${gpu} --num-data=${tox21_num_data}
- snapshot=`ls ${out_dir}/snapshot_iter_* | head -1`
- python inference_tox21.py --in-dir ${out_dir} --gpu ${gpu} --trainer-snapshot ${snapshot} --num-data=${tox21_num_data}
# Tox21 classification task with all labels
out_dir=all_${method}
python train_tox21.py --method ${method} --conv-layers 1 --gpu ${gpu} --epoch 1 --unit-num 10 --out ${out_dir} --batchsize 16 --num-data=${tox21_num_data}
- python inference_tox21.py --in-dir ${out_dir} --num-data=${tox21_num_data}
python predict_tox21_with_classifier.py --in-dir ${out_dir} --num-data=${tox21_num_data}
- snapshot=`ls ${out_dir}/snapshot_iter_* | head -1`
- python inference_tox21.py --in-dir ${out_dir} --gpu ${gpu} --trainer-snapshot ${snapshot} --num-data=${tox21_num_data}
cd ../
# QM9
@@ -58,7 +52,6 @@ done
cd tox21
# BalancedSerialIterator test with Tox21
python train_tox21.py --method nfp --label NR-AR --conv-layers 1 --gpu ${gpu} --epoch 1 --unit-num 10 --out nr_ar_nfp_balanced --iterator-type balanced --eval-mode 0 --num-data 1000
-python inference_tox21.py --in-dir nr_ar_nfp_balanced --gpu ${gpu} --num-data 1000
# ROCAUCEvaluator test with Tox21
python train_tox21.py --method nfp --label NR-AR --conv-layers 1 --gpu ${gpu} --epoch 1 --unit-num 10 --out nr_ar_nfp_balanced --iterator-type serial --eval-mode 1 --num-data 1000
cd ..
