pytorch怎么修改网络的权重或阈值参数

model.state_dict()['layer1.0.weight'].copy_(w1)

w1是自己提前设置好的数，把layer1.0.weight修改为w1，w1需要是tensor格式