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 · 342 阅读 · 0 评论 -
torch-crf
安装命令行:>>>pip install pytorch-crf使用:from torchcrf import CRF原创 2020-08-25 10:11:38 · 2832 阅读 · 14 评论 -
pytorch中tensor类型转换
转载自:https://www.jb51.net/article/177575.htm,做简要记录1、tensor与numpy数组:# tensor转numpytensor.numpy()# numpy转tensortorch.from_numpy()2、tensor与list:# tensor转listtensor.tolist()# list转tensortorch.tensor()3、tensor类型转换tensor = torch.Tensor(3, 5) # torch.转载 2020-08-21 13:13:13 · 2066 阅读 · 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 · 147 阅读 · 0 评论
分享