pytorch | 利用batch normalization对Variable进行normalize/instance normalize

本文介绍了在PyTorch中如何利用_BatchNorm子类实现Batch Normalization和Instance Normalization。通过示例展示了创建和使用这两种规范化层的方法,包括在训练和测试模式下的设置,并强调了调整输入维度以实现Instance Normalization的重要性。
摘要由CSDN通过智能技术生成

天啦噜!!我发现更新的pytorch已经有instance normalization了!!
不用自己折腾了!!
-2017.5.25


利用 nn.Module 里的 子类 _BatchNorm (在torch.nn.modules.batchnorm中定义),可以实现各种需求的normalize。
这里写图片描述
在docs里,可以看到,有3种normalization layer,但其实他们都是继承了_BatchNorm这个类的,所以我们看看BatchNorm2d,就可以对其他的方法举一反三啦~

先来看看文档
这里写图片描述

不清楚没关系,接下来用例子讲解:
创建一个BatchNorm2d的实例的方法如下


                
评论 5
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值