- 博客(2)
- 收藏
- 关注
RSS订阅原创 Pytorch模型在eval模式下的输入和输出问题
Pytorch训练好的模型在eval模式下，输入数据有一个强制要求： 输入的数据Size第一个数必须是样本的数量，例如输入的是一个三通道64x64的样本，则其Size属性必须为： torch.Size([1, 3, 64, 64]) 否则会报错输入与模型维数不对应。 当输入包含多个样本时，其Size第一个数也应该是样本的数量。 对于多分类输出可以用 torch.max() 来确定最终输出的label，该函数第一个参数为torch，第二个参数用于指定坐标轴，0为行，1为列。返回指定行（列）.
2021-08-16 16:31:00
543
原创 Pytorch保存和加载模型
Pytorch模型的保存和加载可以使用： torch.save() 和 torch.load() 该对函数有两种使用方法： 一，直接传入模型，进而保存整个模型 二，对传入的模型使用 .state_dict() 方法，提取模型的参数进而只保存参数。只保存参数需要记住模型的结构，在加载时先搭好模型结构再利用 .load_state_dic() 方法加载参数。 完整示例如下： #模型搭建 net = vgg() #模型训练 optimizer = torch.optim.SG.
2021-08-16 16:17:09
95
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人