change 2 last full connected layers to 1 softmax layer