fix fully connected layer in cnn_model
remove unnecessary code