PyTorch中eval模式和train模式

在训练前写model.train(),进入训练模式。
在预测前写model.eval(),进入预测模式。

参考https://blog.csdn.net/weixin_43593330/article/details/107547202

posted @ 2020-11-29 19:36  火锅先生  阅读(1916)  评论(0编辑  收藏  举报