correction in convnetwork when there is no hidden layer after convolution