fix mushroom.go output layer size