2019年8月5日
PyTorch中的Batch Normalization
摘要： Pytorch中的BatchNorm的API主要有： 一般来说pytorch中的模型都是继承nn.Module类的，都有一个属性trainning指定是否是训练状态，训练状态与否将会影响到某些层的参数是否是固定的，比如BN层或者Dropout层。通常用model.train()指定当前模型model 阅读全文
posted @ 2019-08-05 15:55 leizhao 阅读(6421) 评论(1) 推荐(0) 编辑