在这种情况下,我尝试了所有常见的步骤,以确保推理代码本身没有随机元素,即模型不应基于输入更改其预测。
Why am I getting different results after saving and loading model weights in pytorch?
在这种情况下,我尝试了所有常见的步骤,以确保推理代码本身没有随机元素,即模型不应基于输入更改其预测。
Why am I getting different results after saving and loading model weights in pytorch?
1条答案
按热度按时间bn31dyow1#
在我的例子中,是数据。模型的输入在每次运行时都在改变。在训练时,我设置了适合训练的随机数据转换。但是,当然,这在我准备数据进行推理时引入了随机性。所以我改变了
致:
这确保了一致性。