MXNetのJSONファイルはどいていますか?

モデルを保存する場所はMXNetのmodel.save()メソッドによって指定することができます。デフォルトでは、モデルは.paramsファイルと.symbol.jsonファイルとして保存され、それぞれモデルパラメーターとモデル構造が保存されます。これらの2つのファイルは指定されたパスに保存されます。

たとえば、以下のコードは、モデルパラメータとモデル構造を指定したパスに保存します。

# 导入模块
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` というファイルを生成します。

コメントを残す 0

Your email address will not be published. Required fields are marked *


广告
広告は10秒後に閉じます。
bannerAds