寒武纪MLU270保存.cambricon格式模型注意事项

头文件

import torch_mlu.core.mlu_model as ct

需要先执行

ct.save_as_cambricon('test') # 保存成寒武纪格式的可执行文件

然后执行

net_mlu = torch.jit.trace(net_mlu, input_mlu, check_trace=False) # 追踪

后才能完成.cambricon格式模型的保存，执行的先后顺序是必须的，且这两条语句缺一不可。