![](https://img-blog.csdnimg.cn/20201014180756922.png?x-oss-process=image/resize,m_fixed,h_64,w_64)
pytorch
文章平均质量分 91
yqy2001
THU
展开
-
Pytorch使用记录
torch.max(input, dim)函数对dim维度求最大值,返回两个值,第一个是max value,第二个是max index。 分类问题计算accuracy: 根据预测结果列表prediction和label列表y计算: torch.eq函数比较两个tensor,返回一个bool类型的tensor,相同的地方为True,不同的地方为False;通过float()函数将其转换成0、1的float类型;mean()函数用来求平均值,即accuracy,计算出来的是一个tensor,item()用来取出原创 2021-07-31 22:14:42 · 270 阅读 · 0 评论 -
torch-crf
安装命令行: >>>pip install pytorch-crf 使用: from torchcrf import CRF原创 2020-08-25 10:11:38 · 2668 阅读 · 14 评论 -
pytorch中tensor类型转换
转载自:https://www.jb51.net/article/177575.htm,做简要记录 1、tensor与numpy数组: # tensor转numpy tensor.numpy() # numpy转tensor torch.from_numpy() 2、tensor与list: # tensor转list tensor.tolist() # list转tensor torch.tensor() 3、tensor类型转换 tensor = torch.Tensor(3, 5) # torch.转载 2020-08-21 13:13:13 · 1869 阅读 · 0 评论 -
pytorch模型保存
转载自:https://www.jianshu.com/p/6ba95579082c 只保存模型参数 # 保存 torch.save(model.state_dict(), '\parameter.pkl') # 加载 model = TheModelClass(...) model.load_state_dict(torch.load('\parameter.pkl')) 保存完整模型 # 保存 torch.save(model, '\model.pkl') # 加载 model = torch.load转载 2020-08-21 12:33:31 · 95 阅读 · 0 评论