Pytroch 学习笔记
收录pytroch的使用方法
ZouJiaYuY
嘎嘎猛学~
展开
-
【Pytorch 笔记】网络可视化 && torchsummary
这里能看见图像大小的变化过程,以及参数量。可以直观的看见网络模型和参数配置。原创 2023-10-19 16:25:38 · 1090 阅读 · 1 评论 -
【Pytorch 学习笔记】def init_weights() 初始化参数
nn.init.constant_() 是 torch.nn 中的用于填充数值的函数,这里用于指定初始化值,还有许多其他函数可用于此。在生成网络 net 时,会指定 net 最初的权重,对于一些预训练好的模型权重,就可以放在这个部分进行加载。在 CNN 中,经常可以看见 init_weights() 函数,它是用来初始化网络参数的。并不是所有的层都能初始化权重的,比如 nn.MaxPool2d(),它是无法初始化的。init_weights(self, m) 中的 m 就是指 net 中的某一层。原创 2023-10-19 15:51:42 · 2933 阅读 · 1 评论