mxnet——模型加载与保存

一、加载模型与pretrain模型network相同 # loading predict module data_shape_G = 96 Batch = namedtuple('Batch',['data']) sym, arg_params, aux_params = mx.model.load_checkpoint(prefix=r"~/meh_cla",epoch=2) train,va
相关文章
相关标签/搜索