如何使PyTorch子模块处于求值模式？

import torch

model.eval()

import torch
import torch.nn as nn

# 创建一个简单的子模块
class MyModule(nn.Module):
    def __init__(self):
        super(MyModule, self).__init__()
        self.linear = nn.Linear(10, 1)

    def forward(self, x):
        return self.linear(x)

# 创建模型并加载权重
model = MyModule()
model.load_state_dict(torch.load('model_weights.pth'))

# 将模型设置为求值模式
model.eval()

# 使用模型进行推理
input_tensor = torch.randn(1, 10)
output = model(input_tensor)
print(output)