分类导航

程序问答发布时间：2022-06-02 发布网站：大佬教程 code.js-code.com

大佬教程收集整理的这篇文章主要介绍了如何将运行统计数据删除到 pyTorch 中 Norm 层的默认值？，大佬教程大佬觉得挺不错的，现在分享给大家，也给大家做个参考。

如何解决如何将运行统计数据删除到 pyTorch 中 Norm 层的默认值？？

开发过程中遇到如何将运行统计数据删除到 pyTorch 中 Norm 层的默认值？的问题如何解决？下面主要结合日常开发的经验，给出你关于如何将运行统计数据删除到 pyTorch 中 Norm 层的默认值？的解决方法建议，希望对你解决如何将运行统计数据删除到 pyTorch 中 Norm 层的默认值？有所启发或帮助；

我在一些图像上训练了模型。现在要拟合类似的数据集但使用另一种颜色，我想加载这个模型，但我也想从 Batchnorm 层中删除所有运行统计数据（将它们设置为默认值，就像完全未经训练一样）。我应该重置哪些参数？简单的模型看起来像这样

import torch
import torch.nn as nn

class Net(nn.ModulE):
    def __init__(self):
        super(Net,self).__init__()
        
        self.conv0 = nn.Conv2d(3,3,padding = 1)
        self.norm = nn.batchnorm2d(3)
        self.conv = nn.Conv2d(3,padding = 1)
    
def forWARD(self,X):
        x = self.conv0(X)
        x = self.norm(X)
        return self.conv(X)

net = Net()
##or for preTrained it will be
##net = torch.load('net.pth')

def drop_to_default():
    for m in net.modules():
        if type(m) == nn.batchnorm2d:
           ####???####

drop_to_default()

解决方法

最简单的方法是在 reset_running_stats() 对象上运行 BatchNorm 方法：

def drop_to_default():
    for m in net.modules():
        if type(m) == nn.batchNorm2d:
           m.reset_running_stats()

下面是这个方法的源代码：

def reset_running_stats(self) -> None:
      if self.track_running_stats:
            # running_mean/running_var/num_batches... are registered at runtime depending
            # if self.track_running_stats is on
          self.running_mean.zero_()  # Zero (neutral) mean
          self.running_var.fill_(1)  # One (neutral) variance
          self.num_batches_tracked.zero_()  # number of batches tracked

您可以看到源代码 here，_NormBase 类。