MXNet中的模型存放位置可以通過(guò)model.save()
方法指定。默認(rèn)情況下,模型將被保存為.params
文件和.symbol.json
文件,分別存儲(chǔ)模型參數(shù)和模型結(jié)構(gòu)。這兩個(gè)文件將被保存在指定的路徑下。
例如,以下代碼將保存模型參數(shù)和模型結(jié)構(gòu)到指定的路徑下:
# 導(dǎo)入模塊
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))
# 初始化模型參數(shù)
net.initialize()
# 保存模型
net.save('path/to/model')
此代碼將生成path/to/model.params
和path/to/model-symbol.json
文件。