torch中train,eval实际对是model.training的控制

FY_2018

于 2021-08-27 22:15:20 发布

阅读量1.5k

点赞数 1

分类专栏：机器学习 python 文章标签： PyTorch 模型训练 eval模式训练模式属性切换

本文链接：https://blog.csdn.net/FY_2018/article/details/119961478

版权

机器学习同时被 2 个专栏收录

47 篇文章 3 订阅

订阅专栏

python

30 篇文章 1 订阅

订阅专栏

模型在默认情况下是train模型，所以model.training为True
模型在eval情况下，model.training为False

#-*- coding:utf-8 -*-
#Author LJB Create on 2021/8/27
# -*- coding: utf-8 -*-
import torch

class TestModel(torch.nn.Module):
    def __init__(self):
        super(TestModel,self).__init__()
    def forward(self,x):
        return x

model = TestModel()

#默认为model.train()
print(model.training) #True

model.eval()
print(model.training) #False