Segfault in running inference code in Pytorch 0.4 on Raspberry Pi 3B

amrit-das / custom_image_classifier_pytorch

PyTorch Based Image Classifier for image prediction and segregation https://medium.com/hardware-interfacing/custom-image-classifier-using-transfer-learning-in-pytorch-framework-c2f7f5155239

14 stars 7 forks source link

Segfault in running inference code in Pytorch 0.4 on Raspberry Pi 3B #4

Closed Ashutosh1995 closed 5 years ago

Ashutosh1995 commented 5 years ago

I am trying to run the inference code of ERFNet to generate the label masks but I am getting seg fault whenever I try to save the predictions. It randomly runs and then gives seg fault. There are total 254 images in the validation set.

Please suggest !!

amrit-das commented 5 years ago

Could you please also post the link to the code so that I can run and see the problem for myself.