PyTorch中eval模式和train模式
在训练前写model.train(),进入训练模式。
在预测前写model.eval(),进入预测模式。
参考https://blog.csdn.net/weixin_43593330/article/details/107547202
在训练前写model.train(),进入训练模式。
在预测前写model.eval(),进入预测模式。
参考https://blog.csdn.net/weixin_43593330/article/details/107547202