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