当参数保存为numpy数组时，如何加载Pytorch模型？

import torch
import numpy as np

class MyModel(torch.nn.Module):
    def __init__(self):
        super(MyModel, self).__init__()
        # 定义模型的层和参数

    def forward(self, x):
        # 定义模型的前向传播逻辑
        return x

model = MyModel()

saved_params = np.load('saved_params.npy', allow_pickle=True).item()
model.load_state_dict(saved_params)

input_data = torch.randn(1, input_size)  # 输入数据示例
output = model(input_data)