Advertisement

pytorch中的模型评估模式对应于设置volatile=True以及不启用requires_grad

阅读量:

前言:

近期在查阅各类代码时,发现存在一些不太理解的地方,因此进行整理归纳。

  1. model.eval() 与 model.train()

  2. volatile参数设置为True或False

  3. requires_grad属性设为True或False

  4. with torch.no_grad():

这几个部分主要用于控制或规定张量在训练或测试阶段是否执行求导操作,以及其他相关限制。

一、 model.eval() model.train()

权威性文件

通过观察红色方框所展示的内容可以得知,model.eval()等同于model.train(False),接下来再对model.train()进行进一步分析。

几层含义:

1.等价关系

model.train() 等同于 model.train(True)

model.eval() 等同

全部评论 (0)

还没有任何评论哟~