给我一段pytorch float32转float16的代码

以下是将 PyTorch 中的 float32 转换为 float16 的代码：

import torch

def convert_to_float16(model):
    for param in model.parameters():
        param.data = param.data.half()
    return model

# 示例
model = torch.nn.Sequential(
    torch.nn.Linear(10, 5),
    torch.nn.ReLU(),
    torch.nn.Linear(5, 1)
)
model = convert_to_float16(model)

这段代码可以将 PyTorch 模型中的所有参数从 float32 转换为 float16。