GPU=$1

python node_generation.py \
--dataset wiki_cs \
--batch_size 8 \
--accumulate_step 32 \
--eval_iteration 10 \
--patience 5 \
--training_iteration 100000 \
--drop_ratio 0.3 0.6 0.95 \
--k 2 \
--workers 24 \
--device ${GPU} 