卷积神经网络(CNN)杂记,model、train、predict模板
和在构建网络模块时必须继承且在构造函数中需要调用的构造函数不需要写反向传播函数,能够利用autograd根据你定义的前向传播函数自动实现反向传播Net中可学习参数可以通过类对象中的获取Net中可学习参数包括自己定义好的网络模块中每一层的权重weight和偏置bias,注意torch会自动随机初始化权重和偏置,可以通过类对象中的查看每一层的权重和偏置。
原创
2023-06-19 21:24:19 ·
342 阅读 ·
0 评论