--num_epochs 1 --train_batch_size 16 --eval_batch_size 16 --learning_rate 2e-5 --mixed_precision fp16
