python experiment.py \
    --data_path "./GEARS/data" \
    --device "cuda:3" \
    --epochs 20 \
    --out_dir $1