帮我写一个pytorch脚本，读入一个预训练模型，并将预训练模型中所有参数名从module.XX改为XX

当你加载预训练模型时，模型的参数名称可能会以`module.XX`的形式表示，这是因为在保存预训练模型时使用了`torch.nn.DataParallel`来进行多GPU训练。你可以使用以下代码将参数名中的`module.`去除：

import torch

# 加载预训练模型
model = torch.hub.load('pytorch/vision:v0.9.0', 'resnet18', pretrained=True)

# 去除参数名中的"module."
state_dict = model.state_dict()
new_state_dict = {}
for key, value in state_dict.items():
    if key.startswith('module.'):
        new_key = key[7:]  # 去除前面的"module."
        new_state_dict[new_key] = value
    else:
        new_state_dict[key] = value

# 更新模型的参数字典
model.load_state_dict(new_state_dict)

这样，加载的预训练模型的参数名称将不再包含`module.`前缀。你可以根据需要进行打印或进一步使用。