python3 main.py --dataset synthetic \
               --id 5 \
               --encoder GIN \
               --base_lambda 1 \
               --start_lambda 1e-3 \
               --learning_rate 5e-3 \
               --flag CL