pytorch 状态字典:state_dict使用详解( 二 )


```
import numpy as np
state_dict = model.state_dict()
for k, v in state_dict.items():
state_dict[k] = v.numpy()
np.save('model.npy', state_dict)
```
这个代码将状态字典中的张量保存为 numpy 数组,并将它们存储在名为 'model.npy' 的文件中 。

推荐阅读