
pytorch
u_7890
这个作者很懒,什么都没留下…
展开
-
Pytorch如何去掉模型的最后一层
new_model = torch.nn.Sequential( *( list(model.children())[:-1] ) )print(new_model)原创 2021-08-24 21:54:59 · 3332 阅读 · 0 评论 -
pytorch 学习(1:introduction)
学习:https://morvanzhou.github.io/tutorials/machine-learning/torch/ 笔记=============================== (1.1)简单引入 ====================================1.torch numpy 相互转化torch_data = torch.from...转载 2018-10-12 14:57:25 · 160 阅读 · 0 评论 -
【转】torch.squeeze
链接:torch.squeeze“先看torch.squeeze() 这个函数主要对数据的维度进行压缩,去掉维数为1的的维度,比如是一行或者一列这种,一个一行三列(1,3)的数去掉第一个维数为一的维度之后就变成(3)行。squeeze(a)就是将a中所有为1的维度删掉。不为1的维度没有影响。a.squeeze(N) 就是去掉a中指定的维数为一的维度。还有一种形式就是b=torch.sque...转载 2018-10-12 14:40:18 · 378 阅读 · 0 评论 -
python 将字符串形式的字典转化为字典
例子a = "{'name' : 'jim', 'sex' : 'male', 'age': 18}"b = eval(a)用函数eval,eval实际上是解释执行python代码转载 2018-10-16 10:14:59 · 4981 阅读 · 0 评论 -
torch中与tensor相关的bug
1.expected type torch.FloatTensor but got torch.cuda.FloatTensor查看了代码,是有个初始化的tensor是在cpu上的,相加的另一个tensor是在cuda上的。修改:增加.cuda()initial_hidden_state = initial_hidden_state.cuda()...原创 2019-03-28 23:13:16 · 179 阅读 · 0 评论