pytorch

1、

测试时不需要正则化:

model.eval()就可以不运行batch norm、dropout等。

再用model.train()可改回训练模式。

原文地址:https://www.cnblogs.com/lxc1910/p/11489945.html