- 博客(2)
- 资源 (1)
- 收藏
- 关注
原创 mxnet学习(9):使用gluon接口读取symbol预训练模型finetune
使用symbol得到的模型或者gluon的hybridize之后的模型包括一个.json文件(网络结构)和.params文件(参数),gluon可以使用net = gluon.SymbolBlock.imports(json, ['data'], params, ctx)导入网络和参数,这样可以进行测试或者进一步训练。但是如果只需要使用模型的其中一部分,比如只需要conv层,去掉所有fc层,或者...
2019-05-07 16:48:22 2212 1
原创 mxnet学习(8):Trainer
class mxnet.gluon.Trainer(params, optimizer, optimizer_params = None, kvstore = ‘device’, compression_params = None, updata_on_kvstore = None)参考:http://mxnet.incubator.apache.org/versions/master/api/...
2019-05-07 16:12:38 1954
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人