[small_batch] lenet change to 5 layers, bug fixed