pytorch

1、

测试时不需要正则化:

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

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