GPU=$1

python node_generation.py \
--dataset pubmed \
--batch_size 64 \
--accumulate_step 16 \
--eval_iteration 10 \
--patience 10 \
--training_iteration 100000 \
--drop_ratio 0.3 0.6 0.95 \
--k 2 \
--workers 24 \
--device ${GPU} 