[small_batch] lenet change to 6 layers