train 2 hiddenlayer 1softmax then finetune