×

签到

分享到微信

打开微信,使用扫一扫进入页面后,点击右上角菜单,

点击“发送给朋友”或“分享到朋友圈”完成分享

【MagicMind—pytorch_build工具问题】使用pytorch_build将pt转换为mm出错 待解决 嘻嘻嘻嘻嘻嘻2023-09-18 14:07:41 回复 1 查看 使用求助
【MagicMind—pytorch_build工具问题】使用pytorch_build将pt转换为mm出错
分享到:

【寒武纪硬件产品型号】MLU370


【MagicMind版本号】:1.0.1

执行命令:./pytorch_build --pytorch_pt ./yolov5_model/yolov5s.pt  --dynamic_shape false (pt来自于https://github.com/ultralytics/yolov5)

【出错信息】

2023-09-18 05:10:53.096760: INFO: /usr/local/neuware/samples/magicmind/mm_build/main.cc:13] 

==================== Parameter Information

--precision<Str>              : 

--calibration<Bool>           : false 

--rpc_server<Str>             : 

--calibration_algo<Str>       : linear 

--file_list<Vec<Str>>         : 

--calibration_data_path<Str>  : 

--random_calib_range<Vec<Float>>: 

--batch_size<Vec<Int>>        : 

--input_dims<Vec<Vec<Int>>>   : 

--input_layout<Vec<Str>>      : 

--input_dtypes<Vec<Str>>      : 

--dynamic_shape<Bool>         : false 

--vars<Vec<Vec<Float>>>       : 

--means<Vec<Vec<Float>>>      : 

--output_layout<Vec<Str>>     : 

--output_dtypes<Vec<Str>>     : 

--mlu_arch<Vec<Str>>          : 

--plugin<Vec<Str>>            : 

--magicmind_model<Str>        : ./model 

--build_config<Str>           : 

--toolchain_path<Str>         : /usr/local/gcc-linaro-6.2.1-2016.11-x86_64_aarch64-linux-gnu/ 

--rgb2bgr<Bool>               : false 

--cluster_num<Vec<Vec<Int>>>  : 

--pytorch_pt<Str>             : ./yolov5s.pt 

--pt_input_dtypes<Vec<Str>>   : FLOAT 

MagicMind: 268435472

2023-09-18 05:10:53.101732: ERROR:  magicmind/translation/pytorch/module_loader.cc:77] PyTorch loads the model failed due to: [enforce fail at inline_container.cc:209] . file not found: archive/constants.pkl

2023-09-18 05:10:53.101781: ERROR:  magicmind/cc/implement/interface_parser_impl.cc:66]  Invalid argument: Your input model is not suppported by pytorch.

2023-09-18 05:10:53.101801: ERROR: /usr/local/neuware/samples/magicmind/mm_build/parser.cc:23] Invalid argument: Your input model is not suppported by pytorch.

Aborted (core dumped)


版权所有 © 2024 寒武纪 Cambricon.com 备案/许可证号:京ICP备17003415号-1
关闭