python3 sst_baseline.py --model_name BERT \
                      --batch_size 1 \
                      --learning_rate 0.0001 \
                      --lmbda 0.1 \
                      --loss MSE \
                      --embedding_op l2 \
                      --exp_num 8 \
                      --normalization l1 \
                      --normalization2 none  \
                      --no-softmax \
                      --cuda \
                      --train