MXNet的json存放位置在哪

avatar
作者
筋斗云
阅读量:6

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.paramspath/to/model-symbol.json文件。

广告一刻

为您即时展示最新活动产品广告消息,让您随时掌握产品活动新动态!