![](https://img-blog.csdnimg.cn/20201014180756930.png?x-oss-process=image/resize,m_fixed,h_64,w_64)
Pytorch
pangpd
Just a boy caught up in dreams and fantasies !
展开
-
【Pytorch】创建data.DataLoader,对于参数pin_memory的理解
锁页内存理解(pinned memory or page locked memory)https://blog.csdn.net/dgh_dean/article/details/53130871What is the disadvantage of using pin_memory: https://discuss.pytorch.org/t/what-is-the-disadvantage-of-using-pin-memory/1702pin_memory就是锁页内存,创建DataLoader时,转载 2020-12-30 09:43:32 · 632 阅读 · 0 评论 -
【智障问题】记录一次智障问题(微笑)
1. 问题描述还原了一个CNN模型,自定了Conv3dRelu 卷积层,但是编译器一直提示unexpected argument ,观察了传入的参数,并没有发现什么问题,又定位了好久一直没有找到原因。2. 问题解决(1) 一开始我以为是pytorch版本的问题,于是转移到笔记上,发现还是这个问题;(2) 又查看了之前写的其他模型,并没有发现类似这种写法有什么问题,于是复制了其他模型的一个层,在新模型中去调用该层。发现没有问题!于是详细对比这者的区别:正确:错误:发现 init 写错原创 2020-09-16 09:05:58 · 907 阅读 · 3 评论 -
【Deep Learning】THOP+torchstat 计算PyTorch模型的FLOPs,问题记录与解决
1. 前言在博客中计算PyTorch模型的FLOPs 一文中,介绍了到了衡量一个深度学习模型大小的指标,尤其是FLOPs,它衡量了一个模型的复杂度。如果计算FLOPs,在下面我们介绍了两款工具。THOP: PyTorch-OpCounter与torchstat源码下载链接:THOP: PyTorch-OpCounter:https://github.com/Lyken17/pytorch-OpCountertorchstat:https://github.com/Swall0w/torchstat原创 2020-09-10 14:45:16 · 3837 阅读 · 3 评论