Skip to content

"Segmentation fault: 11" error when using pretrained dataset #79

@YannRoussel

Description

@YannRoussel

Hello,

I am trying to use pre-trained models downloaded following this guide but I get the "Segmentation fault: 11" error even if I use a dummy dataset (i.e. a numpy array of the right shape filled with 0)

I use this script:

d = torch.load("morphoclass/dvc/training/checkpoints-alt-neurites/pc-L6-all-image-tmd-proj-xgb-stratified-k-fold/checkpoint.chk")
model = d["all"]["model"]

dummy = np.zeros((100, 10000))
model.predict(dummy)

I also tried with another model (pc-L5-all-image-tmd-rd-xgb-stratified-k-fold) and with an actual dataset of persistence images but get the same error.

Thank you for your help,

Yann Roussel.

Metadata

Metadata

Labels

bugSomething isn't working

Type

No type

Projects

No projects

Milestone

No milestone

Relationships

None yet

Development

No branches or pull requests

Issue actions