Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

[Training] linux ondevice training with onnxruntim failed by loading checkpoint: Segmentation fault (core dumped) #21918

Closed
scofild429 opened this issue Aug 29, 2024 · 0 comments
Labels
training issues related to ONNX Runtime training; typically submitted using template

Comments

@scofild429
Copy link

Describe the issue

After ondevice training artifacts have been created by python, I want use them for ondevice training in linux with C/C++. I started with CPU provider. For inferencing with corresponding ONNX, everything works. But Segmentation fault (core dumped) comes when I want to load the checkpoint for ondevice training, for C

g_ort_training_api->LoadCheckpoint(checkpoint_path.c_str(), &state_c);

also C++
auto state_cpp = Ort::CheckpointState::LoadCheckpoint(checkpoint_path);
I got the same this issue.

And many onnxruntime version(17,18,19) have been tried, problem remains the same.

To reproduce

download onnx runtime-linux-x64-gpu*, and header file and lib in system, and call the checkpoint loading function in a CMake project or single file project.

Urgency

urgency!!!

ONNX Runtime Installation

Built from Source

ONNX Runtime Version or Commit ID

18

PyTorch Version

2.3.1+cu121

Execution Provider

Default CPU

Execution Provider Library Version

No response

@scofild429 scofild429 added the training issues related to ONNX Runtime training; typically submitted using template label Aug 29, 2024
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
training issues related to ONNX Runtime training; typically submitted using template
Projects
None yet
Development

No branches or pull requests

1 participant