pytorch

1、

测试时不需要正则化:

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

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

posted @ 2019-09-09 09:45  bloglxc  阅读(112)  评论(0编辑  收藏  举报