分类：MxNet

2020
09-29

MxNet预训练模型到Pytorch模型的转换方式

预训练模型在不同深度学习框架中的转换是一种常见的任务。今天刚好DPN预训练模型转换问题，顺手将这个过程记录一下。核心转换函数如下所示：defconvert_from_mxnet(model,checkpoint_prefix,debug=False):_,mxnet_weights,mxnet_aux=mxnet.model.load_checkpoint(checkpoint_prefix,0)remapped_state={}forstate_keyinmodel.state_dict().keys():k=state_key.split('.')aux=Falsemxnet_key=''ifk[0... 继续阅读 >