python main.py \
  --out_dir $1 \
  --data_root ./datasets \
  --batch_size 64 \
  --lr 0.04