pytorch
jayckwang
这个作者很懒,什么都没留下…
展开
-
conda太慢?,pip(离线)安装pytorch指定版本
我们在安装pytorch的时候,往往会进入pytorch官网,接着点击Get Started,然后会出现以下界面:然而官网在一直更新,conda命令倒是好记,但是conda安装速度实在感人,于是就想用pip安装,当pip镜像源设为国内的,如清华,阿里时,速度会很快,然而官网pip安装命令也一直在变,尤其时想安装指定版本的torch还有与之配套的Cuda,电脑环境时,命令行一直搜不到。在网上几经...原创 2019-10-12 18:17:55 · 798 阅读 · 0 评论 -
在训练Pytorch ssd时出现的问题
1.The shape of the mask [32, 8732] at index 0 does not match the shape of the indexed tensor [279424, 1] at index 0张量大小不匹配,解决办法:在multibox_loss.py的98 loss_c[pos] = 0 # filter out pos boxes for ...原创 2019-07-16 11:32:35 · 4108 阅读 · 5 评论 -
RuntimeError: invalid argument 0: Sizes of tensors must match except in dimension 0. Got 312 and 224
在使用pytorch的transorm进行图片resize时,只设定一个参数size,transforms.Resize(size),结果报题目中的错误。这是由transforms.Resize()造成的:该函数的参数设置成tuple格式即可transforms.Resize((width,height))...原创 2019-10-07 21:07:12 · 537 阅读 · 0 评论