GPU=$1

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