![](https://img-blog.csdnimg.cn/20201014180756913.png?x-oss-process=image/resize,m_fixed,h_64,w_64)
pytorch
文章平均质量分 75
Mirocle007
这个作者很懒,什么都没留下…
展开
-
对于pytorch中张量的理解
对于pytorch中tensor的理解1. tensor的组成2. 代码测试 1. tensor的组成 一切都是源于torch.is_storage这个函数的挖掘,因为不懂storage是什么,所以去查了,结果大吃一惊。 tensor分为头信息区(Tensor)和存储区(Storage),信息区主要存放以下信息:形状(size)、步长(stride)、数据类型(type)等。而真正的数据则保存成连续数组,保存在存储区。具体结构见下图(引用自参考1): 2. 代码测试 In [15]: import to原创 2020-09-07 23:33:42 · 745 阅读 · 0 评论 -
pytorch1.2.0下使用目标检测模型训练报错"RuntimeError: expected device cpu and dtype Float but got device cuda"解决办法
pytorch1.2.0下使用目标检测模型训练报错"RuntimeError: expected device cpu and dtype Float but got device cuda"解决办法 报错信息: RuntimeError: expected device cpu and dtype Float but got device cuda 报错原因: 报错可能是代码中存在de...原创 2020-02-28 15:37:04 · 4272 阅读 · 2 评论 -
pytorch1.2.0下使用目标检测模型训练报错"RuntimeError: received 0 items of ancdata"解决办法
pytorch1.2.0下使用目标检测模型训练报错"RuntimeError: received 0 items of ancdata"解决办法 报错信息: RuntimeError: received 0 items of ancdata 报错原因: 报错可能是因为文件描述器共享策略达到了峰值(leaking FDs) 解决办法一: 设置加载数据的 num_workers=0,...原创 2019-10-22 21:51:08 · 4266 阅读 · 1 评论