"Segmentation fault: 11" error when using pretrained dataset

Hello,

I am trying to use pre-trained models downloaded following this [guide](https://bbpgitlab.epfl.ch/ml/user/casalegn/morphoclass-dvc-userguide) 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.