memo for training CUDA_VISIBLE_DEVICES=6 python -m blindness.main train --config_path blindness/configs/base_aug.json