GPU=$1

python node_generation.py \
--dataset arxiv \
--batch_size 16 \
--accumulate_step 64 \
--eval_iteration 50 \
--training_iteration 100000 \
--drop_ratio 0.7 0.8 0.9 1.0  \
--hid_dim 512 \
--k 2 \
--patience 100 \
--workers 24 \
--device ${GPU}