Use the PyTorch Model Predict function block in the matlab2024a simulink module to import a neural network trained using pytorch.
조회 수: 10 (최근 30일)
이전 댓글 표시
大家好,我现在准备用matlab2024a,simulink中的PyTorch Model Predict来导入在pytorch中训练好的神经网络输入到simulink中做机器人动力学控制。我现在的环境配置是:matlab2024a,conda环境的python环境,解释器版本3.9,pytorch版本采用的 cpu版本,神经网络也已经保存为cpu版本。
在matlab命令行检查python:
>>pyenv
ans =
PythonEnvironment - 属性:
Version: "3.9"
Executable: "C:\Users\64375\.conda\envs\conda_env_39\python.exe"
Library: "C:\Users\64375\.conda\envs\conda_env_39\python39.dll"
Home: "C:\Users\64375\.conda\envs\conda_env_39"
Status: Loaded
ExecutionMode: InProcess
ProcessID: "21020"
ProcessName: "MATLAB"
在simulink模块中按照官方的教程导入神经网络policy_net_matlab.pth文件,使用函数torch.load(). 输入维度与输出维度设置与网络参数一致。没有加载前处理与后处理函数。因为用不到。点击运行报如下错误,如图片所示:
댓글 수: 0
답변 (0개)
참고 항목
카테고리
Help Center 및 File Exchange에서 Big Data Processing에 대해 자세히 알아보기
Community Treasure Hunt
Find the treasures in MATLAB Central and discover how the community can help you!
Start Hunting!