
PyTorch
justtoomuchforyou
没人在乎。
展开
-
torch.mul, mm, matmul, bmm, broadcast乘法机制
torch.mm只能让两个二维tensor作矩阵乘法 torch.mul作element-wise的矩阵点乘,维数不限,可以矩阵乘标量 torch.bmm作batch单位的矩阵乘法,维度只能为3。当第0维维数不等时报错,但可用matmul相乘 torch.matmul 正常的矩阵乘法运算,两个输入必须都是Tensor...原创 2020-04-21 22:23:47 · 2662 阅读 · 2 评论 -
整理pytorch报错
Stack Overflow[1]中有这个问题的描述,自己写了一个dice loss,没法反向传播,报这个错。原因应该是因为用了torch.argmax(),好像这个函数不可导,所以没法反向传播。但是...原创 2020-03-10 22:54:43 · 9246 阅读 · 8 评论 -
RuntimeError: CUDA error: device-side assert triggered
模型训练着突然报错,没找到问题在哪,把这个错简单复现了一下,网上都说是torch.nn.functional的cross_entropy的input和target上出了问题,target的index不对什么的[1,2],如下:target是[1,2,3]和[2,3,0]都没问题,[1,2,100]和[99,100,101]就不对,还没搞明白,先记一下。[1]https://blog.csdn...原创 2019-12-29 14:58:05 · 6260 阅读 · 0 评论 -
AttributeError: module 'scipy.misc' has no attribute 'imread'
AttributeError: module 'scipy.misc' has no attribute 'imread'。查阅网上资料之后发现,大部分解决办法都是说没有安装PIL第三方库,库名现在为Pillow,推荐直接使用命令pip install Pillow进行安装,但是我的问题并没有因此得到解决,还是继续报错AttributeError: module 'scipy.misc' ha...转载 2019-12-14 15:29:22 · 285 阅读 · 0 评论 -
Linux服务器上装Anaconda和PyTorch
安装Anacondawget 要装的anaconda版本,anaconda2带的是Python2.7,anaconda3带的是3.7,不要选错了。然后bash 这个.sh文件进行安装。如果anaconda装好了还想装其他的Python版本,比如conda create --name python37 python=3.7,这样装Python3.7。然后执行activate python37...原创 2019-11-10 00:56:24 · 1187 阅读 · 0 评论 -
pytorch DataLoader [Errno 22] invalid argument以及[Errno 32] Broken pipe
在跑网上的pytorch代码时,总是遇到同一个问题,在调用DataLoader时总会报错[Errno 22] invalid argument,通常还会由enumerate()带出,然后下一项是w.start(),网上找不到好的解决方法,好像是和Windows系统并发操作有关系,我的解决办法就是把其它所有打开的Python项目shut down,包括PyCharm和jupyter noteboo...原创 2019-08-28 11:55:07 · 816 阅读 · 1 评论 -
pytorch整型tensor调用Dropout导致kernel自动重启
遇到一个奇怪的问题运行上面的代码,导致kernel直接重启,如下图,而且没有报错将arange(0, 12)改成浮点型arange(0., 12.)可以正常Dropout,输出结果。或者写成:input = t.arange(0, 12).float().view(3, 4)添加一个.float(),也没问题。目前只在Dropout上发现这个问题,其他方法没试过。...原创 2019-09-23 22:38:18 · 328 阅读 · 0 评论 -
numpy中repeat和tile用法,区别于torch.repeat()
numpy中repeat和tile用法:原博:https://blog.youkuaiyun.com/henryghx/article/details/16906235torch.repeat()文档见:https://pytorch.org/docs/stable/tensors.html?highlight=repeat#torch.Tensor.repeatPyTorch中的r...原创 2019-09-24 21:08:28 · 3894 阅读 · 0 评论