定义函数，将训练好的模型转换为 Inference 模型

好的，以下是将训练好的模型转换为Inference模型的函数定义：

import torch

def convert_to_inference_model(model_path):
    # Load the trained model
    model = torch.load(model_path)

    # Set the model to evaluation mode
    model.eval()

    # Create a new model object without the training-related stuff
    inference_model = type(model)()

    # Copy over the parameters from the trained model to the inference model
    inference_model.load_state_dict(model.state_dict())

    # Return the inference model
    return inference_model

该函数将加载训练好的模型，将其设置为评估模式，然后创建一个新的模型对象，并将训练模型的参数复制到新模型中，最后返回新模型。通过这种方式，我们可以快速创建一个适用于推理的模型对象。