PT模型文件：修改参数值

import torch
import torch.nn as nn


#下载模型参数
net = torch.load('yolov5n.pt')#再加载网络的参数
print(net)
#detect = net['model'].model[-1].m
print(net['model'].model[-1].m)  # 数字是索引号，英文是index

detect[0] = nn.Conv2d(64, 75, kernel_size=(1, 1), stride=(1, 1))
detect[1] = nn.Conv2d(128, 75, kernel_size=(1, 1), stride=(1, 1))
detect[2] = nn.Conv2d(256, 75, kernel_size=(1, 1), stride=(1, 1))

print(net['model'].model[-1])
#按参数名修改权重
torch.save(net, './yolov5n_voc.pt')