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 · 3497 阅读 · 0 评论 -
detectron2环境搭建及训练数据
文章目录背景配置环境搭建步骤下载源码创建虚拟环境安装torch,torchvision,cudatoolkit编译detecron2源码测试demo背景最近在跑mask-rcnn,看到facebook推荐使用detectron2,所以就去配置了一下detecron2的环境,虽然二者的核心代码是差不多的,但是明显detectron2环境的配置过程比较简单,所以下面就来看看如何配置detecron2的环境吧~配置搭建环境之前,先来说一下我的显卡配置,毕竟不同的显卡所支持的cuda版本是不一样的,比如GT原创 2021-11-29 21:30:31 · 2123 阅读 · 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 · 1086 阅读 · 0 评论
分享