深度学习
LZZ and MYY
嵌入式Linux、工控、虚拟化
展开
-
WIN10 图神经网络Pytorch搭建
遇到了pytorch geometric安装不上的情况,思来想去,可能是版本对不上? 目前的版本:pytorch 1.8.1 cuda 10.2 遇到问题:原创 2021-05-29 20:52:42 · 124 阅读 · 0 评论 -
Pytorch中nn.Moudle模块和nn.functional模块库的不同
nn.Moudle模块库中存放着很多和神经网络相关的工具,而nn.functional里也一样提供了很多网络层相关的函数功能。 最重要的不同点在于,nn.functional里定义的网络层不可自动学习参数,还需要使用nn.Parameter封装。nn.functional设计的初衷是为了一些不需要学习参数的层,比如激活层,BN层这些,这样就不用在nn.Moudle里定义了。 ...原创 2021-05-24 12:17:57 · 236 阅读 · 0 评论