pytorch
文章平均质量分 57
鹰立如睡
社会会以相同程度的你对自己的尊重,给予它对你的尊重。
展开
-
如何查看bert等模型的梯度,用pytorch
print("=============更新之前===========")temp = 0 #控制打印的参数个数for name, parms in model.named_parameters(): temp += 1 if temp == 2: break print('-->name:', name) print('-->para:', parms) print('-->grad_requirs:', parms.requ.原创 2021-10-31 11:00:18 · 637 阅读 · 0 评论 -
AssertionError:torch not compiled with cuda enabled
anaconda里除了有python包之外(我们下载python可不就下载的这个包吗),还集成了很多数据分析用的包,不用我们再一一安装了。如果在path里配置了anaconda的环境变量,anaconda prompt 和 cmd并无太大差别,只不过anaconda prompt更方便,更好用,有很多快捷键。anaconda创建环境,激活,就可以在这个环境中下载需要的包的版本,即使与别的环境的版本冲突问题也被anaconda解决了,就这样用anaconda来管理不同的python环境。在微信读书原创 2021-01-11 14:11:30 · 13943 阅读 · 2 评论 -
RuntimeError: expected backend CPU and dtype Float but got backend CPU and dtype Long
源代码import torchx = torch.tensor([5])#初始化权重参数w,偏移量b,设置自动求导w = torch.randn(1,requires_grad=True)b = torch.randn(1,requires_grad=True)# 实现前向传播y = torch.mul(w,x)z = torch.add(y,b)print("x,w,b的require_grad属性分别为{},{},{}".format(x.requires_grad,w.requ原创 2020-08-08 12:26:31 · 2008 阅读 · 0 评论 -
win10 cpu版本的 pytorch 安装记录
墙内在官网下载太慢了,下了一下午,一直报错Could not find a version that satisfies the requirement torch==1.4.0最后在这个cpu版本的pytorch安装博客里找到了安装方法,成功安装,用的是国内的镜像网站下载的torch 和 torch vision...原创 2020-08-05 16:57:02 · 193 阅读 · 0 评论