tensorflow网络训练不需要bn层 pytorch需要