BN层,这里进行验证:
定义三个模型:
定义模型1 : 一层卷积层和一层BN层网络
import numpy as np
import math
import torch
import torch.nn...m.running_mean.data.normal_(0, 1)
m.running_var.data.uniform_(1, 2)
定义模型2 : 一个卷积层网络...padding=1, bias=False)
def forward(self, x):
x = self.conv1(x)
return x
定义模型3 : 一层卷积网络...这里手动计算模型2的卷积过程,然后和模型2输出进行对比。...卷积原理如图
模型2有8个卷积核,每个kernel尺度为(3,3,3)对应待卷积特征图(C,H,W),因为pad=1,stride=1,卷积之后输出特征图尺度为(1,8,64,64),首先对输出进行填充