python infer.py
python train_model.py
python infer.py --mask_attr
python infer.py --mask_attr --batch_size 32
python run_finetune.py --mask_attr --batch_size 64
python run_finetune_masking.py --mask_attr --batch_size 80