python3 main.py --dataset synthetic \
               --id 5 \
               --encoder GIN \
               --learning_rate 5e-3 \
               --epochs 500 \
               --save_model True \
               --flag base