MXNet模型参数的存储与读取

1.读写NDArray

通过 nd.save(’filename‘，ndType) 和 nd.load(’filename‘，ndType) 存储和读取

from mxnet import nd
from mxnet.gluon import nn

x = nd.ones(3)
nd.save('x', x)
x2 = nd.load('x')
x2 # 得到的x2是一个二维数组，而x是一维数组

# 存储两个向量
y = nd.zeros(4)
nd.save('xy', [x,y])
x2, y2 = nd.load('xy')
(x2, y2)

# 甚至可以存储一个字符串映射到NDArray的字典
mydict = {'x':x, 'y':y}
nd.save('mydict',mydict)
mydict2 = nd.load('mydict')
mydict2['x']

2.读写Gluon模型参数

通过Block类提供的 save_parameters 和 load_parameters 实现模型读写

class MLP(nn.Block):
    def __init__(self, **kwargs):