zl程序教程

您现在的位置是:首页 >  后端

当前栏目

【转载】 pytorch自定义网络结构不进行参数初始化会怎样?

PyTorch 进行 参数 自定义 转载 怎样 初始化 网络结构
2023-09-11 14:19:20 时间

原文地址:

https://blog.csdn.net/u011668104/article/details/81670544

 

 

 

----------------------------------------------------------------------------------------------------------------

 

 

 

答案:不怕不怕,pytorch自己默认有初始化

 

证据如下:

    torch.nn.Conv2d的参数初始化
    https://pytorch.org/docs/stable/_modules/torch/nn/modules/conv.html#Conv2d
   

Conv2d继承自_ConvNd,在_ConvNd中,可以看到默认参数就是进行初始化的,如下图所示

 

 

 

 

 

 

 

 

 

 

 

 

 

 

2)    torch.nn.BatchNorm2d

 

同理

 

 

 

 

 

 

 

 

 

 

 

3)torch.nn.Linear


同上