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

• 模型在默认情况下是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