pytorch提取参数

model.state_dict()["layer1.0.weight"]

b1 = model.state_dict()['layer1.0.bias']

posted @ 2022-09-27 09:12  李里力离  阅读(37)  评论(0)    收藏  举报