训练好的模型怎么加入代码

将训练好的模型保存为文件，通常是以`.pt`或`.ckpt`为后缀名的文件。在代码中加载模型时，可以使用PyTorch提供的`torch.load()`函数来加载模型。例如：

import torch

# 加载模型
model = torch.load('path/to/model.pt')

# 使用模型进行推理
output = model(input)

需要注意的是，加载模型时需要与训练时使用的硬件环境相同，例如如果训练时使用了GPU，则加载模型时也需要使用GPU。此外，还需要确保加载的模型与代码中定义的模型结构相同。