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
相關文章
相關標籤/搜索