python train.py \
 --model gpt \
 --n_layers 6 \
 --n_embd 384 \
 --n_head 6 \
 --n_train 200000 \
 --n_test 20000 \
 --batch_size 256 \
 --dataset graph \
 --deg 2 \
 --path 5 \
 --num_nodes 50 \
 --lr 0.0001 \
 --use_top \
 --compile
