MXNet中的模型存放位置可以通过model.save()
方法指定。默认情况下,模型将被保存为.params
文件和.symbol.json
文件,分别存储模型参数和模型结构。这两个文件将被保存在指定的路径下。
例如,以下代码将保存模型参数和模型结构到指定的路径下:
# 导入模块
import mxnet as mx
# 定义模型
net = mx.gluon.nn.Sequential()
with net.name_scope():
net.add(mx.gluon.nn.Dense(10, activation='relu'))
net.add(mx.gluon.nn.Dense(2))
# 初始化模型参数
net.initialize()
# 保存模型
net.save('path/to/model')
此代码将生成path/to/model.params
和path/to/model-symbol.json
文件。