![](https://img-blog.csdnimg.cn/20201014180756927.png?x-oss-process=image/resize,m_fixed,h_64,w_64)
Pytorch
文章平均质量分 59
介绍Pytorch
爱幻想-hjyp
遇到坎就得迈过去
展开
-
Pytorch中named_parameters、named_children、named_modules函数
目录named_parameters函数named_children()函数named_modules()函数例子 named_parameters函数 以迭代器的方式返回model中所有的参数,返回值是一个字典:包含参数的名称和数值大小; 内部实现时使用了递归算法,所以对于嵌套的网络参数,会递归遍历,输出最底层的参数,参看下面的例子; named_children()函数 该函数用来输出网络中的第一层模块名称和实例对象,只会展示最上层的模块名; named_modules()函数 该函数用来递归原创 2022-04-23 11:43:38 · 2933 阅读 · 0 评论 -
detectron2环境搭建及训练数据
文章目录背景配置环境搭建步骤下载源码创建虚拟环境安装torch,torchvision,cudatoolkit编译detecron2源码测试demo 背景 最近在跑mask-rcnn,看到facebook推荐使用detectron2,所以就去配置了一下detecron2的环境,虽然二者的核心代码是差不多的,但是明显detectron2环境的配置过程比较简单,所以下面就来看看如何配置detecron2的环境吧~ 配置 搭建环境之前,先来说一下我的显卡配置,毕竟不同的显卡所支持的cuda版本是不一样的,比如GT原创 2021-11-29 21:30:31 · 1853 阅读 · 0 评论 -
pytorch中的各种乘法
题目Python中的乘法参考链接 Python中的乘法 *乘: 就是普通的主元素相乘 torch.dot(): 点乘 torch.mul:支持矩阵和标量的点乘 torch.mm: 矩阵乘积 torch.matmul: 广播形式下的矩阵乘积 torch.bmm: 可以执行两个batch的乘积 图片来自参考链接 参考链接 pytorch中的各种乘 pytorch中mm和bmm的区别 ...原创 2022-03-16 22:27:30 · 945 阅读 · 0 评论