pytorch 训练和预测时,相同的图片数据,得到的结果不同?
解答:
1、验证、测试和预测时,要 加上 model.eval()
2、数据预处理不一致,训练时数据做了归一化、标准化等操作,预测时也要做相同的操作
3、未设置随机数种子
1、pytorch模型保存是基于pickle保存的,是将模型object对象保存了,这个保存在加载的时候,就要求你当前环境和你保存模型时候的环境完全一样(pytorch版本、自己定义的class类之类的)
Input type (torch.FloatTensor) and weight type (torch.cuda.FloatTensor) should be the same
此处要注释掉,原因不明
modelscope问题:
win10 coda python 3.7